diff --git a/nmdc_automation/models/nmdc.py b/nmdc_automation/models/nmdc.py index e0fb46dd..fdd5cbf9 100644 --- a/nmdc_automation/models/nmdc.py +++ b/nmdc_automation/models/nmdc.py @@ -9,9 +9,20 @@ import yaml -from nmdc_schema.nmdc import DataGeneration, FileTypeEnum, MagsAnalysis, MetagenomeAnnotation, MetagenomeAssembly, \ - MetatranscriptomeAnnotation, MetatranscriptomeAssembly, MetatranscriptomeExpressionAnalysis, NucleotideSequencing, \ - ReadBasedTaxonomyAnalysis, ReadQcAnalysis, WorkflowExecution +from nmdc_schema.nmdc import ( + DataGeneration, + MagsAnalysis, + MetagenomeAnnotation, + MetagenomeAssembly, + MetagenomeSequencing, + MetatranscriptomeAnnotation, + MetatranscriptomeAssembly, + MetatranscriptomeExpressionAnalysis, + NucleotideSequencing, + ReadBasedTaxonomyAnalysis, + ReadQcAnalysis, + WorkflowExecution +) import nmdc_schema.nmdc as nmdc @@ -32,6 +43,7 @@ def workflow_process_factory(record: Dict[str, Any], validate: bool = False) -> "nmdc:MagsAnalysis": MagsAnalysis, "nmdc:MetagenomeAnnotation": MetagenomeAnnotation, "nmdc:MetagenomeAssembly": MetagenomeAssembly, + "nmdc:MetagenomeSequencing": MetagenomeSequencing, "nmdc:MetatranscriptomeAnnotation": MetatranscriptomeAnnotation, "nmdc:MetatranscriptomeAssembly": MetatranscriptomeAssembly, "nmdc:MetatranscriptomeExpressionAnalysis": MetatranscriptomeExpressionAnalysis,