ensure relative to /data

Author: tylbees

projects/fal/src/fal/app.py

@@ -811,7 +811,10 @@ async def lifespan(self, app: fastapi.FastAPI):
             sys.path.insert(0, "")
 
         for directory in self.host_kwargs.get("data_dirs", []):
-            warm_dir(directory)
+            warm_path = Path(directory).expanduser()
+            if not warm_path.is_absolute():
+                warm_path = Path("/data") / warm_path
+            warm_dir(os.fspath(warm_path))
 
         _print_python_packages()
         setup_started_at = time.perf_counter()

projects/fal/tests/unit/test_app.py

@@ -402,6 +402,38 @@ def setup(self):
     ]
 
 
+@pytest.mark.asyncio
+async def test_app_lifespan_resolves_relative_data_dirs_from_data(
+    isolate_agent_env, monkeypatch: pytest.MonkeyPatch
+):
+    import fal.app as fal_app_module
+
+    calls: list[tuple[str, str | None]] = []
+
+    monkeypatch.setattr(
+        fal_app_module,
+        "warm_dir",
+        lambda directory: calls.append(("warm", str(directory))),
+    )
+
+    class WarmedApp(App):
+        data_dirs = ["models", "nested/tokenizer"]
+
+        def setup(self):
+            calls.append(("setup", None))
+
+    app = WarmedApp()
+
+    async with app.lifespan(FastAPI()):
+        pass
+
+    assert calls == [
+        ("warm", "/data/models"),
+        ("warm", "/data/nested/tokenizer"),
+        ("setup", None),
+    ]
+
+
 def test_wrap_app_allows_resolver_with_container_kind():
     from fal.app import wrap_app