Move rebuild jobs index

Author: deep1401

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "transformerlab"
-version = "0.0.31"
+version = "0.0.32"
 description = "Python SDK for Transformer Lab"
 readme = "README.md"
 requires-python = ">=3.10"

src/lab/experiment.py

@@ -116,6 +116,9 @@ def create_job(self):
         new_job = Job.create(new_job_id)
         new_job.set_experiment(self.id)
 
+        # Update jobs index now that a new job exists
+        self.rebuild_jobs_index()
+
         return new_job
 
     def get_jobs(self, type: str = "", status: str = ""):
@@ -124,11 +127,6 @@ def get_jobs(self, type: str = "", status: str = ""):
         type: If not blank, filter by jobs with this type.
         status: If not blank, filter by jobs with this status.
         """
-
-        # Rebuild the index
-        # TODO: The point of the index is to not do this every time
-        self.rebuild_jobs_index()
-
         # First get jobs of the passed type
         job_list = []
         if type:
@@ -145,7 +143,6 @@ def get_jobs(self, type: str = "", status: str = ""):
                 job_json = job.get_json_data()
             except Exception:
                 continue
-
             # Filter for status
             if status and (job_json.get("status", "") != status):
                 continue
@@ -157,7 +154,7 @@ def get_jobs(self, type: str = "", status: str = ""):
             # If it passed filters then add as long as it has job_data
             if "job_data" in job_json:
                 results.append(job_json)
-
+        
         return results
 
     ###############################