@@ -30,7 +30,7 @@ def _run_async(coro):
3030 # Use the async version (a_* methods) instead
3131 raise RuntimeError (
3232 "Cannot use sync method when already in async context. "
33- "Use the async version instead (e.g., await lab.a_save_artifact () instead of lab.save_artifact())."
33+ "Use the async version instead (e.g., await lab.async_save_artifact () instead of lab.save_artifact())."
3434 )
3535 except RuntimeError as e :
3636 # Check if this is our custom error or a "no running loop" error
@@ -279,7 +279,7 @@ async def async_get_checkpoint_to_resume(self) -> Optional[str]:
279279 return None
280280
281281 # Build the checkpoint path from parent job's checkpoints directory
282- checkpoint_path = await self .a_get_parent_job_checkpoint_path (parent_job_id , checkpoint_name )
282+ checkpoint_path = await self .async_get_parent_job_checkpoint_path (parent_job_id , checkpoint_name )
283283
284284 # Verify the checkpoint exists
285285 if checkpoint_path and await storage .exists (checkpoint_path ):
@@ -292,7 +292,7 @@ def get_parent_job_checkpoint_path(self, parent_job_id: str, checkpoint_name: st
292292 Get the full path to a checkpoint from a parent job (sync version).
293293
294294 This is a sync wrapper around the async implementation.
295- Use a_get_parent_job_checkpoint_path () if you're already in an async context.
295+ Use async_get_parent_job_checkpoint_path () if you're already in an async context.
296296 """
297297 return _run_async (self .async_get_parent_job_checkpoint_path (parent_job_id , checkpoint_name ))
298298
@@ -379,7 +379,7 @@ def save_artifact(
379379 Save an artifact file or directory into this job's artifacts folder (sync version).
380380
381381 This is a sync wrapper around the async implementation.
382- Use a_save_artifact () if you're already in an async context.
382+ Use aasync_save_artifact () if you're already in an async context.
383383 """
384384 return _run_async (self .async_save_artifact (source_path , name , type , config ))
385385
@@ -456,7 +456,7 @@ async def async_save_artifact(
456456 is_image = config ["is_image" ]
457457
458458 # Use the existing save_dataset method
459- output_path = await self .a_save_dataset (
459+ output_path = await self .async_save_dataset (
460460 df = df ,
461461 dataset_id = dataset_id ,
462462 additional_metadata = additional_metadata if additional_metadata else None ,
@@ -1005,7 +1005,7 @@ async def async_save_model(
10051005 config ["parent_model" ] = parent_model
10061006
10071007 # Use save_artifact with type="model"
1008- return await self .a_save_artifact (
1008+ return await self .async_save_artifact (
10091009 source_path = source_path ,
10101010 name = name ,
10111011 type = "model" ,
@@ -1153,7 +1153,7 @@ def get_artifacts_dir(self) -> str:
11531153 Get the artifacts directory path for the current job (sync version).
11541154
11551155 This is a sync wrapper around the async implementation.
1156- Use a_get_artifacts_dir () if you're already in an async context.
1156+ Use async_get_artifacts_dir () if you're already in an async context.
11571157 """
11581158 return _run_async (self .async_get_artifacts_dir ())
11591159
@@ -1185,7 +1185,7 @@ def get_artifact_paths(self) -> list[str]:
11851185 Get list of artifact file paths for the current job (sync version).
11861186
11871187 This is a sync wrapper around the async implementation.
1188- Use a_get_artifact_paths () if you're already in an async context.
1188+ Use async_get_artifact_paths () if you're already in an async context.
11891189 """
11901190 return _run_async (self .async_get_artifact_paths ())
11911191
0 commit comments