void updateArgsBeforeExport()

in src/main/java/com/amazonaws/services/neptune/profiles/neptune_ml/NeptuneMLSourceDataModel.java [103:121]

• 16 lines of code
• 7 McCabe index (conditional complexity)

        void updateArgsBeforeExport(Args args, Collection<TrainingDataWriterConfigV2> trainingJobWriterConfigCollection) {
            args.removeOptions("--format");
            args.addOption("--format", RdfExportFormat.ntriples.name());

            boolean exportEdgesOnly = true;

            for (TrainingDataWriterConfigV2 trainingDataWriterConfigV2 : trainingJobWriterConfigCollection) {
                for (LabelConfigV2 labelConfig : trainingDataWriterConfigV2.nodeConfig().getAllClassificationSpecifications()) {
                    String taskType = labelConfig.taskType();
                    if (taskType.equals(RdfTaskTypeV2.classification.name()) || taskType.equals(RdfTaskTypeV2.regression.name())){
                        exportEdgesOnly = false;
                    }
                }
            }

            if (!args.contains("--rdf-export-scope") && exportEdgesOnly){
                args.addOption("--rdf-export-scope", RdfExportScope.edges.name());
            }
        }