Fix typos

wandb/sdk/wandb_watch.py

@@ -70,7 +70,7 @@ def watch(
     log_gradients = log in {"gradients", "all"}
     log_parameters = log in {"parameters", "all"}
 
-    _watch(
+    return _watch(
         models,
         criterion,
         log_gradients=log_gradients,
@@ -90,11 +90,14 @@ def _watch(
     idx: Optional[int] = None,
     log_graph: bool = False,
 ) -> List["wandb.wandb_torch.TorchGraph"]:
+    assert wandb.run is not None, "You must call `wandb.init` before calling watch"
+
     global _global_watch_idx
 
     graphs = []
 
     idx = idx or _global_watch_idx
+
     for local_idx, model in enumerate(models):
         global_idx = idx + local_idx
         _global_watch_idx += 1