From 6ad96a7cc358b1df2cb3cffd7c78627bce4f8d69 Mon Sep 17 00:00:00 2001
From: Sidharth Hulyalkar <sidharth.hulyalkar@datajoint.com>
Date: Wed, 1 Nov 2023 13:22:33 -0500
Subject: [PATCH] remove model_id fetch

---
 element_facemap/facemap_train.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/element_facemap/facemap_train.py b/element_facemap/facemap_train.py
index 44d85fe..1a2dd2c 100644
--- 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