wandb 0.17.8rc1__py3-none-any.whl → 0.18.0__py3-none-any.whl

wandb/integration/openai/fine_tuning.py

@@ -37,23 +37,23 @@ from openai.types.fine_tuning.fine_tuning_job import (  # noqa: E402
 np = util.get_module(
     name="numpy",
     required="`numpy` not installed >> This integration requires numpy!  To fix, please `pip install numpy`",
-    lazy="False",
+    lazy=False,
 )
 
 pd = util.get_module(
     name="pandas",
     required="`pandas` not installed >> This integration requires pandas!  To fix, please `pip install pandas`",
-    lazy="False",
+    lazy=False,
 )
 
 
 class WandbLogger:
     """Log OpenAI fine-tunes to [Weights & Biases](https://wandb.me/openai-docs)."""
 
-    _wandb_api: wandb.Api = None
+    _wandb_api: Optional[wandb.Api] = None
     _logged_in: bool = False
-    openai_client: OpenAI = None
-    _run: Run = None
+    openai_client: Optional[OpenAI] = None
+    _run: Optional[Run] = None
 
     @classmethod
     def sync(

wandb/integration/tensorboard/log.py

@@ -6,7 +6,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 import wandb
 import wandb.util
 from wandb.sdk.lib import telemetry
-from wandb.viz import custom_chart
+from wandb.sdk.lib.viz import custom_chart
 
 if TYPE_CHECKING:
     import numpy as np
@@ -54,9 +54,6 @@ def make_ndarray(tensor: Any) -> Optional["np.ndarray"]:
 def namespaced_tag(tag: str, namespace: str = "") -> str:
     if not namespace:
         return tag
-    elif tag in namespace:
-        # This happens with tensorboardX
-        return namespace
     else:
         return namespace + "/" + tag
 

wandb/jupyter.py

@@ -167,8 +167,6 @@ def notebook_metadata_from_jupyter_servers_and_kernel_id():
             urljoin(s["url"], "api/sessions"), params={"token": s.get("token", "")}
         ).json()
         for nn in res:
-            # TODO: wandb/client#400 found a case where res returned an array of
-            # strings...
             if isinstance(nn, dict) and nn.get("kernel") and "notebook" in nn:
                 if nn["kernel"]["id"] == kernel_id:
                     return {
@@ -176,7 +174,24 @@ def notebook_metadata_from_jupyter_servers_and_kernel_id():
                         "path": nn["notebook"]["path"],
                         "name": nn["notebook"]["name"],
                     }
-    return None
+
+    if not kernel_id:
+        return None
+
+    # Built-in notebook server in VS Code
+    try:
+        from IPython import get_ipython
+
+        ipython = get_ipython()
+        notebook_path = ipython.kernel.shell.user_ns.get("__vsc_ipynb_file__")
+        if notebook_path:
+            return {
+                "root": os.path.dirname(notebook_path),
+                "path": notebook_path,
+                "name": os.path.basename(notebook_path),
+            }
+    except Exception:
+        return None
 
 
 def notebook_metadata(silent: bool) -> Dict[str, str]: