updates

src/axolotl/telemetry/__init__.py

@@ -1,5 +1,15 @@
 """Init for axolotl.telemetry module."""
 
-from .manager import ModelConfig, TelemetryConfig, TelemetryManager, init_telemetry_manager
+from .manager import (
+    ModelConfig,
+    TelemetryConfig,
+    TelemetryManager,
+    init_telemetry_manager,
+)
 
-__all__ = ["TelemetryConfig", "TelemetryManager", "ModelConfig", "init_telemetry_manager"]
+__all__ = [
+    "TelemetryConfig",
+    "TelemetryManager",
+    "ModelConfig",
+    "init_telemetry_manager",
+]

src/axolotl/telemetry/manager.py

@@ -126,8 +126,7 @@ class TelemetryManager:
 
         base_model = base_model.lower()
         return any(
-            org.lower() in base_model 
-            for org in self.whitelist.get("organizations", [])
+            org.lower() in base_model for org in self.whitelist.get("organizations", [])
         )
 
     def _init_posthog(self):
@@ -192,7 +191,7 @@ class TelemetryManager:
                         "run_id": self.run_id,
                         "system_info": system_info,
                         **properties,
-                    }
+                    },
                 )
             except Exception as e:
                 logger.warning(f"Failed to send telemetry event: {e}")
@@ -266,4 +265,4 @@ class TelemetryManager:
 
 def init_telemetry_manager() -> TelemetryManager:
     """Initialize telemetry system"""
-    return TelemetryManager(TelemetryConfig())
+    return TelemetryManager(TelemetryConfig())

src/axolotl/telemetry/whitelist.yaml

@@ -8,4 +8,4 @@ organizations:
   - "NousResearch"
   - "allenai"
   - "amd"
-  - "tiiuae"
+  - "tiiuae"