wandb 0.13.10__py3-none-any.whl → 0.14.0__py3-none-any.whl

wandb/wandb_agent.py

@@ -121,6 +121,47 @@ class AgentProcess:
         return self._proc.terminate()
 
 
+def _create_sweep_command_args(command: Dict) -> Dict[str, Any]:
+    """Create various formats of command arguments for the agent.
+
+    Raises:
+        ValueError: improperly formatted command dict
+
+    """
+    if "args" not in command:
+        raise ValueError('No "args" found in command: %s' % command)
+    # four different formats of command args
+    # (1) standard command line flags (e.g. --foo=bar)
+    flags: List[str] = []
+    # (2) flags without hyphens (e.g. foo=bar)
+    flags_no_hyphens: List[str] = []
+    # (3) flags with false booleans ommited  (e.g. --foo)
+    flags_no_booleans: List[str] = []
+    # (4) flags as a dictionary (used for constructing a json)
+    flags_dict: Dict[str, Any] = {}
+    for param, config in command["args"].items():
+        _value: Any = config.get("value", None)
+        if _value is None:
+            raise ValueError('No "value" found for command["args"]["%s"]' % param)
+        _flag: str = f"{param}={_value}"
+        flags.append("--" + _flag)
+        flags_no_hyphens.append(_flag)
+        if isinstance(_value, bool):
+            # omit flags if they are boolean and false
+            if _value:
+                flags_no_booleans.append("--" + param)
+        else:
+            flags_no_booleans.append("--" + _flag)
+        flags_dict[param] = _value
+    return {
+        "args": flags,
+        "args_no_hyphens": flags_no_hyphens,
+        "args_no_boolean_flags": flags_no_booleans,
+        "args_json": [json.dumps(flags_dict)],
+        "args_dict": flags_dict,
+    }
+
+
 class Agent:
     POLL_INTERVAL = 5
     REPORT_INTERVAL = 0
@@ -169,8 +210,11 @@ class Agent:
             os.environ["WANDB_DIR"] = os.path.abspath(os.getcwd())
 
     def is_flapping(self):
-        """Flapping occurs if the agents receives FLAPPING_MAX_FAILURES non-0
-        exit codes in the first FLAPPING_MAX_SECONDS"""
+        """Determine if the process is flapping.
+
+        Flapping occurs if the agents receives FLAPPING_MAX_FAILURES non-0 exit codes in
+        the first FLAPPING_MAX_SECONDS.
+        """
         if os.getenv(wandb.env.AGENT_DISABLE_FLAPPING) == "true":
             return False
         if time.time() < wandb.START_TIME + self.FLAPPING_MAX_SECONDS:
@@ -183,7 +227,6 @@ class Agent:
         )
 
     def run(self):  # noqa: C901
-
         # TODO: catch exceptions, handle errors, show validation warnings, and make more generic
         sweep_obj = self._api.sweep(self._sweep_id, "{}")
         if sweep_obj:
@@ -340,50 +383,9 @@ class Agent:
 
         return response
 
-    @staticmethod
-    def _create_command_args(command: Dict) -> Dict[str, Any]:
-        """Create various formats of command arguments for the agent.
-
-        Raises:
-            ValueError: improperly formatted command dict
-
-        """
-        if "args" not in command:
-            raise ValueError('No "args" found in command: %s' % command)
-        # four different formats of command args
-        # (1) standard command line flags (e.g. --foo=bar)
-        flags: List[str] = []
-        # (2) flags without hyphens (e.g. foo=bar)
-        flags_no_hyphens: List[str] = []
-        # (3) flags with false booleans ommited  (e.g. --foo)
-        flags_no_booleans: List[str] = []
-        # (4) flags as a dictionary (used for constructing a json)
-        flags_dict: Dict[str, Any] = {}
-        for param, config in command["args"].items():
-            _value: Any = config.get("value", None)
-            if _value is None:
-                raise ValueError('No "value" found for command["args"]["%s"]' % param)
-            _flag: str = f"{param}={_value}"
-            flags.append("--" + _flag)
-            flags_no_hyphens.append(_flag)
-            if isinstance(_value, bool):
-                # omit flags if they are boolean and false
-                if _value:
-                    flags_no_booleans.append("--" + param)
-            else:
-                flags_no_booleans.append("--" + _flag)
-            flags_dict[param] = _value
-        return {
-            "args": flags,
-            "args_no_hyphens": flags_no_hyphens,
-            "args_no_boolean_flags": flags_no_booleans,
-            "args_json": [json.dumps(flags_dict)],
-            "args_dict": flags_dict,
-        }
-
     @staticmethod
     def _create_sweep_command(command: Optional[List] = None) -> List:
-        """Returns sweep command, filling in environment variable macros."""
+        """Return sweep command, filling in environment variable macros."""
         # Start from default sweep command
         command = command or Agent.DEFAULT_SWEEP_COMMAND
         for i, chunk in enumerate(command):
@@ -442,7 +444,7 @@ class Agent:
 
         env = dict(os.environ)
 
-        sweep_vars: Dict[str, Any] = Agent._create_command_args(command)
+        sweep_vars: Dict[str, Any] = _create_sweep_command_args(command)
 
         if "${args_json_file}" in sweep_command:
             with open(json_file, "w") as fp:
@@ -588,11 +590,9 @@ def run_agent(
 
 
 def agent(sweep_id, function=None, entity=None, project=None, count=None):
-    """
-    Generic agent entrypoint, used for CLI or jupyter.
+    """Run a function or program with configuration parameters specified by server.
 
-    Will run a function or program with configuration parameters specified
-    by server.
+    Generic agent entrypoint, used for CLI or jupyter.
 
     Arguments:
         sweep_id: (dict) Sweep ID generated by CLI or sweep API

wandb/wandb_controller.py

@@ -87,7 +87,7 @@ def _id_generator(size=10, chars=string.ascii_lowercase + string.digits):
 
 
 class ControllerError(Exception):
-    """Base class for sweep errors"""
+    """Base class for sweep errors."""
 
     pass
 
@@ -128,7 +128,6 @@ class _WandbController:
     """
 
     def __init__(self, sweep_id_or_config=None, entity=None, project=None):
-
         # sweep id configured in constuctor
         self._sweep_id: Optional[str] = None
 
@@ -327,7 +326,7 @@ class _WandbController:
             self._create["parameters"][name]["b"] = b
 
     def configure_controller(self, type):
-        """configure controller to local if type == 'local'."""
+        """Configure controller to local if type == 'local'."""
         self._configure_check()
         self._create.setdefault("controller", {})
         self._create["controller"].setdefault("type", type)

wandb/wandb_run.py

@@ -1,5 +1,4 @@
-"""
-Compatibility wandb_run module.
+"""Compatibility wandb_run module.
 
 In the future use:
     from wandb.sdk.wandb_run import Run

wandb/wandb_torch.py

@@ -16,7 +16,7 @@ torch = None
 
 
 def nested_shape(array_or_tuple, seen=None):
-    """Figures out the shape of tensors possibly embedded in tuples
+    """Figure out the shape of tensors possibly embedded in tuples
     i.e
     [0,0] returns (2)
     ([0,0], [0,0]) returns (2,2)

wandb/xgboost/__init__.py

@@ -1,5 +1,4 @@
-"""
-Compatibility xgboost module.
+"""Compatibility xgboost module.
 
 In the future use:
     from wandb.integration.xgboost import wandb_callback

{wandb-0.13.10.dist-info → wandb-0.14.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wandb
-Version: 0.13.10
+Version: 0.14.0
 Summary: A CLI and library for interacting with the Weights and Biases API.
 Home-page: https://github.com/wandb/wandb
 Author: Weights & Biases
@@ -25,7 +25,7 @@ Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: Click (!=8.0.0,>=7.0)
-Requires-Dist: GitPython (>=1.0.0)
+Requires-Dist: GitPython (!=3.1.29,>=1.0.0)
 Requires-Dist: requests (<3,>=2.0.0)
 Requires-Dist: psutil (>=5.0.0)
 Requires-Dist: sentry-sdk (>=1.0.0)
@@ -55,6 +55,7 @@ Requires-Dist: minio ; extra == 'kubeflow'
 Requires-Dist: google-cloud-storage ; extra == 'kubeflow'
 Requires-Dist: sh ; extra == 'kubeflow'
 Provides-Extra: launch
+Requires-Dist: awscli ; extra == 'launch'
 Requires-Dist: nbconvert ; extra == 'launch'
 Requires-Dist: nbformat ; extra == 'launch'
 Requires-Dist: chardet ; extra == 'launch'
@@ -62,7 +63,10 @@ Requires-Dist: iso8601 ; extra == 'launch'
 Requires-Dist: typing-extensions ; extra == 'launch'
 Requires-Dist: boto3 ; extra == 'launch'
 Requires-Dist: botocore ; extra == 'launch'
+Requires-Dist: google-auth ; extra == 'launch'
+Requires-Dist: google-cloud-compute ; extra == 'launch'
 Requires-Dist: google-cloud-storage ; extra == 'launch'
+Requires-Dist: google-cloud-artifact-registry ; extra == 'launch'
 Requires-Dist: kubernetes ; extra == 'launch'
 Provides-Extra: media
 Requires-Dist: numpy ; extra == 'media'