update workflow_process_factory

microbiomedata · Nov 20, 2024 · 0e81082 · 0e81082
1 parent f6a52f0
commit 0e81082
Showing 1 changed file with 15 additions and 3 deletions.
diff --git 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,