Merge pull request #23 from transformerlab/fix/lattice-machine-workspace

Add a variable to return the correct workspace directory when the env var is set

Author: deep1401

pyproject.toml

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

src/lab/dirs.py

@@ -27,6 +27,11 @@ def set_organization_id(organization_id: str | None) -> None:
 
 
 def get_workspace_dir() -> str:
+    # Remote SkyPilot workspace override (highest precedence)
+    # Only return container workspace path when value is exactly "true"
+    if os.getenv("_TFL_REMOTE_SKYPILOT_WORKSPACE") == "true":
+        return "/workspace"
+
     # Explicit override wins
     if "TFL_WORKSPACE_DIR" in os.environ:
         value = os.environ["TFL_WORKSPACE_DIR"]