move part table insert after insert into facemapProcessing

datajoint · Nov 1, 2023 · bc4f6d6 · bc4f6d6
1 parent 8281f6a
commit bc4f6d6
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/element_facemap/facemap_train.py b/element_facemap/facemap_train.py
@@ -425,13 +425,6 @@ def make(self, key):
             model_id, refined_model_name, model_description, model_output_path
         )
 
-        self.RetrainedModelFile.insert1(
-            dict(
-                train_model_time=train_model_time,
-                retrain_file=model_output_path,
-            ),
-        )
-
         train_model_time = datetime.fromtimestamp(
             model_output_path.stat().st_mtime
         ).strftime("%Y-%m-%d %H:%M:%S")
@@ -442,3 +435,10 @@ def make(self, key):
                 train_model_time=train_model_time,
             )
         )
+
+        self.RetrainedModelFile.insert1(
+            dict(
+                train_model_time=train_model_time,
+                retrain_file=model_output_path,
+            ),
+        )