Fix pytests that were failling locally on MacOS (#407)

Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>

Author: abidlabs

.changeset/small-camels-stare.md

@@ -0,0 +1,5 @@
+---
+"trackio": patch
+---
+
+feat:Fix pytests that were failling locally on MacOS

tests/unit/test_run.py

@@ -1,10 +1,11 @@
 import time
+from datetime import datetime
 from unittest.mock import MagicMock, patch
 
 import pytest
 
 from trackio import Run, init
-from trackio.utils import _cached_whoami
+from trackio.sqlite_storage import SQLiteStorage
 
 
 class DummyClient:
@@ -39,6 +40,7 @@ def test_init_resume_modes(temp_dir):
     assert run.name == "new-run"
 
     run.log({"x": 1})
+    SQLiteStorage.bulk_log("test-project", "new-run", [{"x": 1}])
     run.finish()
 
     run = init(
@@ -84,10 +86,10 @@ def test_init_resume_modes(temp_dir):
     assert run.name == "nonexistent-run"
 
 
-@patch("huggingface_hub.whoami")
+@patch("trackio.utils._cached_whoami")
 @patch("time.time")
-def test_run_name_generation_with_space_id(mock_time, mock_whoami, temp_dir):
-    mock_whoami.return_value = {"name": "testuser"}
+def test_run_name_generation_with_space_id(mock_time, mock_cached_whoami, temp_dir):
+    mock_cached_whoami.return_value = {"name": "testuser"}
     mock_time.return_value = 1234567890
 
     client = DummyClient()
@@ -100,8 +102,6 @@ def test_run_name_generation_with_space_id(mock_time, mock_whoami, temp_dir):
     )
     assert run.name == "testuser-1234567890"
 
-    _cached_whoami.cache_clear()
-
 
 def test_reserved_config_keys_rejected(temp_dir):
     with pytest.raises(ValueError, match="Config key '_test' is reserved"):
@@ -113,9 +113,9 @@ def test_reserved_config_keys_rejected(temp_dir):
         )
 
 
-@patch("huggingface_hub.whoami")
-def test_automatic_username_and_timestamp_added(mock_whoami, temp_dir):
-    mock_whoami.return_value = {"name": "testuser"}
+@patch("trackio.utils._cached_whoami")
+def test_automatic_username_and_timestamp_added(mock_cached_whoami, temp_dir):
+    mock_cached_whoami.return_value = {"name": "testuser"}
 
     run = Run(
         url="http://test",
@@ -128,13 +128,9 @@ def test_automatic_username_and_timestamp_added(mock_whoami, temp_dir):
     assert "_Created" in run.config
     assert run.config["learning_rate"] == 0.01
 
-    from datetime import datetime
-
     created_time = datetime.fromisoformat(run.config["_Created"])
     assert created_time.tzinfo is not None
 
-    _cached_whoami.cache_clear()
-
 
 def test_run_group_added(temp_dir):
     run = Run(

tests/unit/test_sqlite_storage.py

@@ -6,10 +6,13 @@
 import sqlite3
 import tempfile
 import time
+from pathlib import Path
 
 import orjson
 import pytest
 
+import trackio.sqlite_storage
+import trackio.utils
 from trackio.sqlite_storage import SQLiteStorage
 
 
@@ -94,12 +97,21 @@ def test_import_export(temp_dir):
 
 
 def _worker_using_sqlite_storage(
-    project, worker_id, duration_seconds=2, sync_start_time=None
+    project, worker_id, duration_seconds=2, sync_start_time=None, temp_dir=None
 ):
     """
     Worker that uses SQLiteStorage methods for database access.
     This will be protected by ProcessLock when available.
     """
+    if temp_dir:
+        os.environ["TRACKIO_DIR"] = temp_dir
+        from pathlib import Path
+
+        import trackio.sqlite_storage
+        import trackio.utils
+
+        trackio.utils.TRACKIO_DIR = Path(temp_dir)
+        trackio.sqlite_storage.TRACKIO_DIR = Path(temp_dir)
 
     def aggressive_get_connection(db_path):
         conn = sqlite3.connect(str(db_path), timeout=0.01)
@@ -144,23 +156,24 @@ def aggressive_get_connection(db_path):
 def test_concurrent_database_access_without_errors():
     """
     Test that concurrent database access doesn't produce 'database is locked' errors.
-    This test should fail on main (without ProcessLock) and pass with ProcessLock fix.
     """
     with tempfile.TemporaryDirectory() as temp_dir:
         os.environ["TRACKIO_DIR"] = str(temp_dir)
+        trackio.utils.TRACKIO_DIR = Path(temp_dir)
+        trackio.sqlite_storage.TRACKIO_DIR = Path(temp_dir)
+
         project = "concurrent_test"
 
         num_processes = 8
         duration = 2
 
-        # Synchronized start time (0.5s from now) to make all processes hit db simultaneously
         sync_start_time = time.time() + 0.5
 
         with multiprocessing.Pool(processes=num_processes) as pool:
             results = [
                 pool.apply_async(
                     _worker_using_sqlite_storage,
-                    (project, i, duration, sync_start_time),
+                    (project, i, duration, sync_start_time, temp_dir),
                 )
                 for i in range(num_processes)
             ]

trackio/__init__.py

@@ -218,10 +218,6 @@ def init(
                 utils.embed_url_in_notebook(space_url)
     context_vars.current_project.set(project)
 
-    client = None
-    if not space_id:
-        client = Client(url, verbose=False)
-
     if resume == "must":
         if name is None:
             raise ValueError("Must provide a run name when resume='must'")
@@ -250,7 +246,7 @@ def init(
     run = Run(
         url=url,
         project=project,
-        client=client,
+        client=None,
         name=name,
         group=group,
         config=config,

trackio/run.py

@@ -31,6 +31,29 @@ def __init__(
         auto_log_gpu: bool = False,
         gpu_log_interval: float = 10.0,
     ):
+        """
+        Initialize a Run for logging metrics to Trackio.
+
+        Args:
+            url: The URL of the Trackio server (local Gradio app or HF Space).
+            project: The name of the project to log metrics to.
+            client: A pre-configured gradio_client.Client instance, or None to
+                create one automatically in a background thread with retry logic.
+                Passing None is recommended for normal usage. Passing a client
+                is useful for testing (e.g., injecting a mock client).
+            name: The name of this run. If None, a readable name like
+                "brave-sunset-0" is auto-generated. If space_id is provided,
+                generates a "username-timestamp" format instead.
+            group: Optional group name to organize related runs together.
+            config: A dictionary of configuration/hyperparameters for this run.
+                Keys starting with '_' are reserved for internal use.
+            space_id: The HF Space ID if logging to a Space (e.g., "user/space").
+                If provided, media files will be uploaded to the Space.
+            auto_log_gpu: Whether to automatically log GPU metrics (utilization,
+                memory, temperature) at regular intervals.
+            gpu_log_interval: The interval in seconds between GPU metric logs.
+                Only used when auto_log_gpu is True.
+        """
         self.url = url
         self.project = project
         self._client_lock = threading.Lock()