Merge pull request #331 from ndif-team/fix-remote-logging

REMOTE_LOGGING toggle takes affect

src/nnsight/intervention/backends/remote.py

@@ -10,7 +10,7 @@
 import requests
 import socketio
 import torch
-from tqdm import tqdm
+from tqdm.auto import tqdm
 
 from ... import __IPYTHON__, CONFIG, remote_logger
 from ...schema.request import RequestModel, StreamValueModel

src/nnsight/schema/config.py

@@ -4,6 +4,8 @@
 import yaml
 from pydantic import BaseModel
 
+from ..logger import remote_logger
+
 
 class ApiConfigModel(BaseModel):
     HOST: str = "ndif.dev"
@@ -22,6 +24,16 @@ class AppConfigModel(BaseModel):
     FRAME_INJECTION: bool = True
     GLOBAL_TRACING: bool = True
 
+    def __setattr__(self, name, value):
+        if name == "REMOTE_LOGGING":
+            self.on_remote_logging_change(value)
+        super().__setattr__(name, value)
+
+    def on_remote_logging_change(self, value: bool):
+        if value != self.REMOTE_LOGGING:
+            remote_logger.disabled = (not value)
+        self.__dict__["REMOTE_LOGGING"] = value
+
 
 class ConfigModel(BaseModel):
     API: ApiConfigModel = ApiConfigModel()