Skip to content

Commit a52a5f9

Browse files
authored
Merge pull request #1205 from transformerlab/fix/sdk-function-names
Fix lab sdk a_ async_ typos
2 parents 839e62b + 495a827 commit a52a5f9

2 files changed

Lines changed: 9 additions & 9 deletions

File tree

lab-sdk/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
44

55
[project]
66
name = "transformerlab"
7-
version = "0.0.63"
7+
version = "0.0.64"
88
description = "Python SDK for Transformer Lab"
99
readme = "README.md"
1010
requires-python = ">=3.10"

lab-sdk/src/lab/lab_facade.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)