Reset telemetry sent

iterative · Sep 13, 2024 · c564916 · c564916
1 parent 606e164
commit c564916
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_telemetry.py b/tests/test_telemetry.py
@@ -1,13 +1,14 @@
+from datachain import telemetry
 from datachain.lib.dc import DataChain
-from datachain.telemetry import is_enabled
 
 
 def test_is_enabled():
-    assert not is_enabled()
+    assert not telemetry.is_enabled()
 
 
 def test_telemetry_api_call(mocker, tmp_dir):
     patch_send = mocker.patch("iterative_telemetry.IterativeTelemetryLogger.send")
+    telemetry._telemetry_sent = False
 
     DataChain.from_storage(tmp_dir.as_uri())
     assert patch_send.call_count == 1