wandb 0.19.1rc1__py3-none-win32.whl → 0.19.3__py3-none-win32.whl

wandb/sdk/service/server_sock.py

@@ -4,6 +4,7 @@ import threading
 import time
 from typing import TYPE_CHECKING, Any, Callable, Dict, Optional
 
+import wandb
 from wandb.proto import wandb_server_pb2 as spb
 from wandb.sdk.internal.settings_static import SettingsStatic
 
@@ -225,7 +226,7 @@ class DebugThread(threading.Thread):
         while True:
             time.sleep(30)
             for thread in threading.enumerate():
-                print(f"DEBUG: {thread.name}")
+                wandb.termwarn(f"DEBUG: {thread.name}")
 
 
 class SocketServer:
@@ -266,7 +267,7 @@ class SocketServer:
                 # socket.shutdown() is a more heavy handed approach to interrupting socket.accept()
                 # in the future we might want to consider a more graceful shutdown which would involve setting
                 # a threading Event and then initiating one last connection just to close down the thread
-                # The advantage of the heavy handed approach is that it doesnt depend on the threads functioning
+                # The advantage of the heavy handed approach is that it does not depend on the threads functioning
                 # properly, that is, if something has gone wrong, we probably want to use this hammer to shut things down
                 self._sock.shutdown(socket.SHUT_RDWR)
             except OSError:

wandb/sdk/service/service.py

@@ -160,7 +160,7 @@ class _Service:
                     service_args.append("--no-observability")
 
                 if core_debug(default="False"):
-                    service_args.append("--debug")
+                    service_args.extend(["--log-level", "-4"])
 
                 exec_cmd_list = []
                 termlog(

wandb/sdk/service/streams.py

@@ -6,13 +6,15 @@ StreamAction: Lightweight record for stream ops for thread safety
 StreamMux: Container for dictionary of stream threads per runid
 """
 
+from __future__ import annotations
+
 import functools
 import multiprocessing
 import queue
 import threading
 import time
 from threading import Event
-from typing import Any, Callable, Dict, List, Optional
+from typing import Any, Callable
 
 import psutil
 
@@ -38,7 +40,7 @@ from ..interface.interface_relay import InterfaceRelay
 class StreamThread(threading.Thread):
     """Class to running internal process as a thread."""
 
-    def __init__(self, target: Callable, kwargs: Dict[str, Any]) -> None:
+    def __init__(self, target: Callable, kwargs: dict[str, Any]) -> None:
         threading.Thread.__init__(self)
         self.name = "StreamThr"
         self._target = target
@@ -51,9 +53,9 @@ class StreamThread(threading.Thread):
 
 
 class StreamRecord:
-    _record_q: "queue.Queue[pb.Record]"
-    _result_q: "queue.Queue[pb.Result]"
-    _relay_q: "queue.Queue[pb.Result]"
+    _record_q: queue.Queue[pb.Record]
+    _result_q: queue.Queue[pb.Result]
+    _relay_q: queue.Queue[pb.Result]
     _iface: InterfaceRelay
     _thread: StreamThread
     _settings: SettingsStatic
@@ -111,7 +113,7 @@ class StreamAction:
     _processed: Event
     _data: Any
 
-    def __init__(self, action: str, stream_id: str, data: Optional[Any] = None):
+    def __init__(self, action: str, stream_id: str, data: Any | None = None):
         self._action = action
         self._stream_id = stream_id
         self._data = data
@@ -133,12 +135,12 @@ class StreamAction:
 
 class StreamMux:
     _streams_lock: threading.Lock
-    _streams: Dict[str, StreamRecord]
-    _port: Optional[int]
-    _pid: Optional[int]
-    _action_q: "queue.Queue[StreamAction]"
+    _streams: dict[str, StreamRecord]
+    _port: int | None
+    _pid: int | None
+    _action_q: queue.Queue[StreamAction]
     _stopped: Event
-    _pid_checked_ts: Optional[float]
+    _pid_checked_ts: float | None
     _mailbox: Mailbox
 
     def __init__(self) -> None:
@@ -152,7 +154,7 @@ class StreamMux:
         self._mailbox = Mailbox()
         self._mailbox.enable_keepalive()
 
-    def _get_stopped_event(self) -> "Event":
+    def _get_stopped_event(self) -> Event:
         # TODO: clean this up, there should be a better way to abstract this
         return self._stopped
 
@@ -192,7 +194,7 @@ class StreamMux:
         self._action_q.put(action)
         action.wait_handled()
 
-    def stream_names(self) -> List[str]:
+    def stream_names(self) -> list[str]:
         with self._streams_lock:
             names = list(self._streams.keys())
             return names
@@ -273,7 +275,7 @@ class StreamMux:
         if self._check_orphaned():
             self._stopped.set()
 
-        poll_exit_responses: List[pb.PollExitResponse] = []
+        poll_exit_responses: list[pb.PollExitResponse] = []
         for probe_handle in probe_handles:
             result = probe_handle.get_probe_result()
             if result:
@@ -281,7 +283,7 @@ class StreamMux:
 
         progress_printer.update(poll_exit_responses)
 
-    def _finish_all(self, streams: Dict[str, StreamRecord], exit_code: int) -> None:
+    def _finish_all(self, streams: dict[str, StreamRecord], exit_code: int) -> None:
         if not streams:
             return
 
@@ -292,8 +294,8 @@ class StreamMux:
         exit_handles = []
 
         # only finish started streams, non started streams failed early
-        started_streams: Dict[str, StreamRecord] = {}
-        not_started_streams: Dict[str, StreamRecord] = {}
+        started_streams: dict[str, StreamRecord] = {}
+        not_started_streams: dict[str, StreamRecord] = {}
         for stream_id, stream in streams.items():
             d = started_streams if stream._started else not_started_streams
             d[stream_id] = stream

wandb/sdk/verify/verify.py

@@ -37,17 +37,17 @@ def print_results(
     else:
         color = "red"
     if isinstance(failed_test_or_tests, str):
-        print(RED_X)
-        print(click.style(failed_test_or_tests, fg=color, bold=True))
+        print(RED_X)  # noqa: T201
+        print(click.style(failed_test_or_tests, fg=color, bold=True))  # noqa: T201
     elif isinstance(failed_test_or_tests, list) and len(failed_test_or_tests) > 0:
-        print(RED_X)
-        print(
+        print(RED_X)  # noqa: T201
+        print(  # noqa: T201
             "\n".join(
                 [click.style(f, fg=color, bold=True) for f in failed_test_or_tests]
             )
         )
     else:
-        print(CHECKMARK)
+        print(CHECKMARK)  # noqa: T201
 
 
 def check_host(host: str) -> bool:
@@ -58,7 +58,7 @@ def check_host(host: str) -> bool:
 
 
 def check_logged_in(api: Api, host: str) -> bool:
-    print("Checking if logged in".ljust(72, "."), end="")
+    print("Checking if logged in".ljust(72, "."), end="")  # noqa: T201
     login_doc_url = "https://docs.wandb.ai/ref/cli/wandb-login"
     fail_string = None
     if api.api_key is None:
@@ -83,7 +83,7 @@ def check_logged_in(api: Api, host: str) -> bool:
 
 def check_secure_requests(url: str, test_url_string: str, failure_output: str) -> None:
     # check if request is over https
-    print(test_url_string.ljust(72, "."), end="")
+    print(test_url_string.ljust(72, "."), end="")  # noqa: T201
     fail_string = None
     if not url.startswith("https"):
         fail_string = failure_output
@@ -91,7 +91,7 @@ def check_secure_requests(url: str, test_url_string: str, failure_output: str) -
 
 
 def check_cors_configuration(url: str, origin: str) -> None:
-    print("Checking CORs configuration of the bucket".ljust(72, "."), end="")
+    print("Checking CORs configuration of the bucket".ljust(72, "."), end="")  # noqa: T201
     fail_string = None
     res_get = requests.options(
         url, headers={"Origin": origin, "Access-Control-Request-Method": "GET"}
@@ -107,7 +107,7 @@ def check_cors_configuration(url: str, origin: str) -> None:
 
 
 def check_run(api: Api) -> bool:
-    print(
+    print(  # noqa: T201
         "Checking logged metrics, saving and downloading a file".ljust(72, "."), end=""
     )
     failed_test_strings = []
@@ -308,7 +308,7 @@ def log_use_download_artifact(
 
 
 def check_artifacts() -> bool:
-    print("Checking artifact save and download workflows".ljust(72, "."), end="")
+    print("Checking artifact save and download workflows".ljust(72, "."), end="")  # noqa: T201
     failed_test_strings: List[str] = []
 
     # test checksum
@@ -365,7 +365,7 @@ def check_artifacts() -> bool:
 
 def check_graphql_put(api: Api, host: str) -> Tuple[bool, Optional[str]]:
     # check graphql endpoint using an upload
-    print("Checking signed URL upload".ljust(72, "."), end="")
+    print("Checking signed URL upload".ljust(72, "."), end="")  # noqa: T201
     failed_test_strings = []
     gql_fp = "gql_test_file.txt"
     f = open(gql_fp, "w")
@@ -410,7 +410,7 @@ def check_graphql_put(api: Api, host: str) -> Tuple[bool, Optional[str]]:
 
 
 def check_large_post() -> bool:
-    print(
+    print(  # noqa: T201
         "Checking ability to send large payloads through proxy".ljust(72, "."), end=""
     )
     descy = "a" * int(10**7)
@@ -461,7 +461,7 @@ def check_large_post() -> bool:
 
 
 def check_wandb_version(api: Api) -> None:
-    print("Checking wandb package version is up to date".ljust(72, "."), end="")
+    print("Checking wandb package version is up to date".ljust(72, "."), end="")  # noqa: T201
     _, server_info = api.viewer_server_info()
     fail_string = None
     warning = False