Merge branch 'basic_tutorial' into fix_local_predict

.github/workflows/docs.yaml

@@ -17,7 +17,11 @@ jobs:
         fetch-depth: 0 # otherwise, you will failed to push refs to dest repo
     - name: install dacapo
       # run: pip install .[docs]
-      run: pip install sphinx-autodoc-typehints sphinx-autoapi sphinx-click sphinx-rtd-theme myst-parser
+      run: pip install sphinx-autodoc-typehints sphinx-autoapi sphinx-click sphinx-rtd-theme myst-parser jupytext ipykernel nbsphinx
+    - name: parse notebooks
+      run: jupytext --to notebook --execute ./docs/source/notebooks/*.py
+    - name: remove notebook scripts
+      run: rm ./docs/source/notebooks/*.py
     - name: Build and Commit
       uses: sphinx-notes/pages@v2
       with:

dacapo/experiments/tasks/losses/affinities_loss.py

@@ -79,10 +79,15 @@ def compute(self, prediction, target, weight):
             weight[:, self.num_affinities :, ...],
         )
 
-        return (
-            torch.nn.BCEWithLogitsLoss(reduction="none")(affs, affs_target)
-            * affs_weight
-        ).mean() + self.lsds_to_affs_weight_ratio * (
-            torch.nn.MSELoss(reduction="none")(torch.nn.Sigmoid()(aux), aux_target)
-            * aux_weight
-        ).mean()
+        if aux.shape[1] == 0:
+            return torch.nn.BCEWithLogitsLoss(reduction="none")(
+                affs, affs_target
+            ).mean()
+        else:
+            return (
+                torch.nn.BCEWithLogitsLoss(reduction="none")(affs, affs_target)
+                * affs_weight
+            ).mean() + self.lsds_to_affs_weight_ratio * (
+                torch.nn.MSELoss(reduction="none")(torch.nn.Sigmoid()(aux), aux_target)
+                * aux_weight
+            ).mean()

docs/source/index.rst

@@ -9,6 +9,7 @@
 
    overview
    install
+   notebooks/minimal_tutorial
    tutorial
    docker
    aws