clean

Author: slekkala1

llama_stack/providers/utils/job_scheduler/config.py

@@ -52,7 +52,6 @@ class CelerySchedulerConfig(SchedulerConfig):
     )
 
 
-# Union type for all scheduler configs with discriminator
 SchedulerConfigUnion = Annotated[
     InlineSchedulerConfig | CelerySchedulerConfig,
     Field(discriminator="type"),

llama_stack/providers/utils/job_scheduler/inline/scheduler.py

@@ -80,7 +80,6 @@ async def _load_jobs_from_storage(self) -> None:
                 job["started_at"] = datetime.fromisoformat(job["started_at"])
             if job.get("completed_at"):
                 job["completed_at"] = datetime.fromisoformat(job["completed_at"])
-            # Deserialize JobStatus enum
             job["status"] = JobStatus(job["status"])
 
             self._jobs[job["job_id"]] = job
@@ -171,8 +170,6 @@ async def _run_job(self, job_id: str) -> None:
                 await self._save_job_to_storage(job)
 
                 # Execute the job based on job_type
-                # This is where job-specific logic would be called
-                # For now, we'll simulate a simple job execution
                 result = await self._execute_job(job)
 
                 # Mark as completed
@@ -215,9 +212,8 @@ async def _execute_job(self, job: dict) -> dict:
         # Check if a custom executor is registered for this job type
         if job_type in self._job_executors:
             executor = self._job_executors[job_type]
-            return await executor(job_data)  # Call the registered executor
+            return await executor(job_data)
 
-        # No executor registered for this job type
         raise ValueError(f"No executor registered for job type: {job_type}")
 
     async def get_job_info(self, job_id: str) -> dict:

src/llama_stack/providers/utils/job_scheduler/inline/scheduler.py

@@ -74,7 +74,6 @@ async def _load_jobs_from_storage(self) -> None:
 
         for value in stored_values:
             job = json.loads(value)
-            # Deserialize datetime strings back to datetime objects
             job["created_at"] = datetime.fromisoformat(job["created_at"])
             if job.get("started_at"):
                 job["started_at"] = datetime.fromisoformat(job["started_at"])