remove model_id fetch

datajoint · Nov 1, 2023 · 6ad96a7 · 6ad96a7
1 parent c22b03a
commit 6ad96a7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/element_facemap/facemap_train.py b/element_facemap/facemap_train.py
@@ -296,8 +296,8 @@ class FacemapModelTraining(dj.Computed):
     class RetrainedModelFile(dj.Part):
         """Stores newly trained models
 
-        Args:
-            dj (_type_): _description_
+        Attributes:
+            FacemapModelTraining (foreign key):
         """
 
         definition = """
@@ -412,7 +412,6 @@ def make(self, key):
         model_output_path = output_dir / f"{refined_model_name}.pth"
         train_model.save_model(model_output_path)
 
-        model_id = (FacemapModelTrainingTask & key).fetch1("model_id")
         model_description = (FacemapModelTrainingTask & key).fetch1("model_description")
 
         # Insert newly trained model results into FacemapModel table