diff --git a/.buildinfo b/.buildinfo
index c37cbac..396d03e 100644
--- a/.buildinfo
+++ b/.buildinfo
@@ -1,4 +1,4 @@
# Sphinx build info version 1
# This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: dfea3737e91c2b1d2c45602070b0e8ef
+config: fbb4a0474d649aef965cde0860227e97
tags: 645f666f9bcd5a90fca523b33c5a78b7
diff --git a/_modules/cellmap_data/dataloader.html b/_modules/cellmap_data/dataloader.html
index e122534..b4403fc 100644
--- a/_modules/cellmap_data/dataloader.html
+++ b/_modules/cellmap_data/dataloader.html
@@ -154,66 +154,15 @@
Source code for cellmap_data.transforms.targets.distance
def forward(self,x:torch.Tensor):"""Forward pass."""# TODO: Need to figure out how to prevent having inaccurate distance values at the edges --> precompute
- # distance = self._transform(x[b, class_ind].nan_to_num(0))
- distance=self._transform(x[b,class_ind])
- distance[x[b,class_ind].isnan()]=torch.nan
- x[b,class_ind]=distance
+ # distance = self._transform(x.nan_to_num(0))
+ distance=self._transform(x)
+ distance[x.isnan()]=torch.nan
+ x=distancereturnx
@@ -426,10 +375,10 @@
Source code for cellmap_data.transforms.targets.distance
def forward(self,x:torch.Tensor):"""Forward pass."""# TODO: Need to figure out how to prevent having inaccurate distance values at the edges --> precompute
- # distance = self._transform(x[b, class_ind].nan_to_num(0))
- distance=self._transform(x[b,class_ind])
- distance[x[b,class_ind].isnan()]=torch.nan
- x[b,class_ind]=distance
+ # distance = self._transform(x.nan_to_num(0))
+ distance=self._transform(x)
+ distance[x.isnan()]=torch.nan
+ x=distancereturnx
diff --git a/_modules/cellmap_data/utils/dtype.html b/_modules/cellmap_data/utils/dtype.html
index b3ebc53..c101f46 100644
--- a/_modules/cellmap_data/utils/dtype.html
+++ b/_modules/cellmap_data/utils/dtype.html
@@ -154,66 +154,15 @@
diff --git a/_modules/cellmap_data/utils/figs.html b/_modules/cellmap_data/utils/figs.html
index 2c78ae9..9b2c5c5 100644
--- a/_modules/cellmap_data/utils/figs.html
+++ b/_modules/cellmap_data/utils/figs.html
@@ -154,66 +154,15 @@
diff --git a/_modules/index.html b/_modules/index.html
index ea56ae6..54d93fa 100644
--- a/_modules/index.html
+++ b/_modules/index.html
@@ -154,66 +154,15 @@
diff --git a/_sources/index.rst b/_sources/index.rst
index 33023ab..534c4ca 100644
--- a/_sources/index.rst
+++ b/_sources/index.rst
@@ -10,7 +10,11 @@ CellMap-Data: the Docs
This library provides a collection of classes and functions for working with cellmap data, specifically for the CellMap project team training machine learning models. The capabilities include loading data from the CellMap ZarrDataset, transforming data, and splitting data into training and validation sets. Functionality is not provided for writing data.
+
+Contents
+==============
.. autosummary::
+
:recursive:
:toctree:
@@ -25,6 +29,9 @@ This library provides a collection of classes and functions for working with cel
cellmap_data.utils
+.. include:: ../../README.md
+ :parser: recommonmark
+
Links
==================
diff --git a/cellmap_data.CellMapDataLoader.html b/cellmap_data.CellMapDataLoader.html
index 75b6897..81f1c8d 100644
--- a/cellmap_data.CellMapDataLoader.html
+++ b/cellmap_data.CellMapDataLoader.html
@@ -44,7 +44,6 @@
-
@@ -107,6 +106,8 @@
+
+