fix: Use generic weave panel when a defined panel cannot be found (#38)

* fix: Use generic weave panel when a defined panel cannot be found

* chore: bump version to 1.0.10

* test: Change references to entity and project

Author: domphan-wandb

examples/workspaces/01_loading_and_cloning_workspaces.py

@@ -2,12 +2,13 @@
 
 import wandb_workspaces.workspaces as ws
 
-# !! edit to your entity and project !!
+# !! edit to your entity, project, and saved view !!
 entity = os.getenv("WANDB_ENTITY")
-project = os.getenv("WANDB_PROJECT")
+project = "workspace-api-demo2"
+saved_view = "oydw4wx21l"
 
 # 1. Load a workspace from URL
-url = "https://wandb.ai/megatruong/workspace-api-demo2?nw=vnizqj6vq3"
+url = f"https://wandb.ai/{entity}/{project}?nw={saved_view}"
 workspace = ws.Workspace.from_url(url)
 
 # 2a. Edit the workspace and save to the same view

pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 license = "Apache-2.0"
 name = "wandb-workspaces"
-version = "0.1.9"
+version = "0.1.10"
 description = "A library for programatically working with the Weights & Biases UI."
 authors = ["Weights & Biases <[email protected]>"]
 readme = "README.md"

tests/test_workspaces.py

@@ -1,20 +1,24 @@
+import os
 import sys
 from typing import Any, Dict, Generic, Type, TypeVar
 
 import pytest
 from polyfactory.factories import DataclassFactory
 from polyfactory.pytest_plugin import register_fixture
-
+import wandb_workspaces.reports.v2.internal as _wr
 import wandb_workspaces.expr
 import wandb_workspaces.reports.v2 as wr
 import wandb_workspaces.workspaces as ws
+from tests.weave_panel_factory import WeavePanelFactory
 from wandb_workspaces.utils.validators import (
     validate_no_emoji,
     validate_spec_version,
     validate_url,
 )
 from wandb_workspaces.workspaces.errors import SpecVersionError, UnsupportedViewError
 
+ENTITY = os.getenv("WANDB_ENTITY")
+
 T = TypeVar("T")
 
 
@@ -140,13 +144,13 @@ def test_filter_expr(expr, spec):
 
 
 def test_load_workspace_from_url():
-    url = "https://wandb.ai/megatruong/workspace-api-demo?nw=vs71wsgdvrz"
+    url = f"https://wandb.ai/{ENTITY}/workspace-api-demo?nw=tkdujz254ke"
     workspace = ws.Workspace.from_url(url)  # noqa: F841
 
 
 @pytest.mark.xfail(reason="Saving to the same workspace is currently bugged")
 def test_save_workspace():
-    workspace = ws.Workspace(entity="megatruong", project="workspace-api-demo")
+    workspace = ws.Workspace(entity=ENTITY, project="workspace-api-demo")
     workspace.save()
     workspace_name = workspace._internal_name
 
@@ -159,7 +163,7 @@ def test_save_workspace():
 
 
 def test_save_workspace_as_new_view():
-    workspace = ws.Workspace(entity="megatruong", project="workspace-api-demo")
+    workspace = ws.Workspace(entity=ENTITY, project="workspace-api-demo")
     workspace.save_as_new_view()
     workspace_name = workspace._internal_name
 
@@ -243,3 +247,24 @@ def test_validate_url(example, should_pass):
     else:
         with pytest.raises(UnsupportedViewError):
             validate_url(example)
+
+
+@pytest.mark.parametrize(
+    "panel_config, should_return_instance",
+    [
+        (
+            WeavePanelFactory.build_summary_table_panel(),
+            wr.interface.WeavePanelSummaryTable,
+        ),
+        (WeavePanelFactory.build_artifact_panel(), wr.interface.WeavePanelArtifact),
+        (
+            WeavePanelFactory.build_artifact_version_panel(),
+            wr.interface.WeavePanelArtifactVersionedFile,
+        ),
+        (WeavePanelFactory.build_run_var_panel(), wr.interface.WeavePanel),
+        (_wr.UnknownPanel(), wr.interface.UnknownPanel),
+    ],
+)
+def test_panel_lookup(panel_config, should_return_instance):
+    panel = wr.interface._lookup_panel(panel_config)
+    assert isinstance(panel, should_return_instance)