Allow users to enable tracker with an environment variable.

IntelLabs · Sep 20, 2023 · d36f08e · d36f08e
1 parent 393a862
commit d36f08e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/mart/datamodules/fiftyone.py b/mart/datamodules/fiftyone.py
@@ -16,9 +16,10 @@
 
 logger = logging.getLogger(__name__)
 try:
-    # Disable the FiftyOne tracker due to the privacy concern.
-    # Users should delete the following line if they intend to be tracked.
-    os.environ["FIFTYONE_DO_NOT_TRACK"] = "1"
+    # Disable the FiftyOne tracker by default due to the privacy concern.
+    # Users need to export FIFTYONE_DO_NOT_TRACK=0 if they intend to be tracked.
+    if os.getenv("FIFTYONE_DO_NOT_TRACK") is None:
+        os.environ["FIFTYONE_DO_NOT_TRACK"] = "1"
     import fiftyone as fo
     import fiftyone.utils.coco as fouc
 except ImportError: