wandb 0.21.4__py3-none-win32.whl → 0.22.0__py3-none-win32.whl

wandb/__init__.py

@@ -10,7 +10,7 @@ For reference documentation, see https://docs.wandb.com/ref/python.
 """
 from __future__ import annotations
 
-__version__ = "0.21.4"
+__version__ = "0.22.0"
 
 
 from wandb.errors import Error

wandb/__init__.pyi

@@ -107,7 +107,7 @@ if TYPE_CHECKING:
     import wandb
     from wandb.plot import CustomChart
 
-__version__: str = "0.21.4"
+__version__: str = "0.22.0"
 
 run: Run | None
 config: wandb_config.Config

wandb/apis/public/__init__.py

@@ -1,3 +1,45 @@
+__all__ = (
+    "Api",
+    "RetryingClient",  # doc:exclude
+    "requests",  # doc:exclude
+    "ArtifactCollection",
+    "ArtifactCollections",
+    "ArtifactFiles",
+    "Artifacts",
+    "ArtifactType",
+    "ArtifactTypes",
+    "RunArtifacts",
+    "Automations",
+    "File",  # doc:exclude
+    "Files",
+    "HistoryScan",  # doc:exclude
+    "SampledHistoryScan",  # doc:exclude
+    "SlackIntegrations",  # doc:exclude
+    "WebhookIntegrations",  # doc:exclude
+    "Job",  # doc:exclude
+    "QueuedRun",  # doc:exclude
+    "RunQueue",  # doc:exclude
+    "RunQueueAccessType",  # doc:exclude
+    "RunQueuePrioritizationMode",  # doc:exclude
+    "RunQueueResourceType",  # doc:exclude
+    "Project",
+    "Projects",
+    "Sweeps",
+    "QueryGenerator",  # doc:exclude
+    "Registry",
+    "BetaReport",
+    "PanelMetricsHelper",  # doc:exclude
+    "PythonMongoishQueryGenerator",  # doc:exclude
+    "Reports",
+    "Run",
+    "Runs",
+    "Sweep",
+    "Member",
+    "Team",
+    "User",
+)
+
+
 from wandb.apis.public.api import Api, RetryingClient, requests
 from wandb.apis.public.artifacts import (
     ArtifactCollection,

wandb/apis/public/runs.py

@@ -88,10 +88,28 @@ RUN_FRAGMENT = """fragment RunFragment on Run {
 
 @normalize_exceptions
 def _server_provides_internal_id_for_project(client) -> bool:
-    """Returns True if the server allows us to query the internalId field for a project.
-
-    This check is done by utilizing GraphQL introspection in the available fields on the Project type.
+    """Returns True if the server allows us to query the internalId field for a project."""
+    query_string = """
+       query ProbeProjectInput {
+            ProjectType: __type(name:"Project") {
+                fields {
+                    name
+                }
+            }
+        }
     """
+
+    # Only perform the query once to avoid extra network calls
+    query = gql(query_string)
+    res = client.execute(query)
+    return "internalId" in [
+        x["name"] for x in (res.get("ProjectType", {}).get("fields", [{}]))
+    ]
+
+
+@normalize_exceptions
+def _server_provides_project_id_for_run(client) -> bool:
+    """Returns True if the server allows us to query the projectId field for a run."""
     query_string = """
        query ProbeRunInput {
             RunType: __type(name:"Run") {
@@ -209,13 +227,13 @@ class Runs(SizedPaginator["Run"]):
             f"""#graphql
             query Runs($project: String!, $entity: String!, $cursor: String, $perPage: Int = 50, $order: String, $filters: JSONString) {{
                 project(name: $project, entityName: $entity) {{
-                    internalId
+                    {"internalId" if _server_provides_internal_id_for_project(client) else ""}
                     runCount(filters: $filters)
                     readOnly
                     runs(filters: $filters, after: $cursor, first: $perPage, order: $order) {{
                         edges {{
                             node {{
-                                {"projectId" if _server_provides_internal_id_for_project(client) else ""}
+                                {"projectId" if _server_provides_project_id_for_run(client) else ""}
                                 ...RunFragment
                             }}
                             cursor
@@ -603,7 +621,7 @@ class Run(Attrs):
             query Run($project: String!, $entity: String!, $name: String!) {{
                 project(name: $project, entityName: $entity) {{
                     run(name: $name) {{
-                        {"projectId" if _server_provides_internal_id_for_project(self.client) else ""}
+                        {"projectId" if _server_provides_project_id_for_run(self.client) else ""}
                         ...RunFragment
                     }}
                 }}