update trainign_status type

stuqdog · stuqdog · commit cc2eaa31e18f · 2023-10-05T09:13:55.000-04:00
diff --git a/src/viam/app/ml_training_client.py b/src/viam/app/ml_training_client.py
@@ -92,7 +92,7 @@ async def get_training_job(self, id: str) -> TrainingJobMetadata:
 
         return response.metadata
 
-    async def list_training_jobs(self, org_id: str, training_status: Optional[TrainingStatus]) -> List[TrainingJobMetadata]:
+    async def list_training_jobs(self, org_id: str, training_status: Optional[TrainingStatus.ValueType]) -> List[TrainingJobMetadata]:
         """Returns training job data for all jobs within an org.
 
         Args:
