added octo pull request to repo, octo-models/octo#61

Toradus · Toradus · commit 2db4c97f24e6 · 2024-03-10T17:32:00.000+01:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ authors = [
     {name = "Marcel Rühle", email="unesl@student.kit.edu"}
 ]
 description = "Uha: Minimal adaption of the dataloader proposed in Octo for loading over 25 embodiments."
-version = "0.2.4"
+version = "0.2.5"
 readme = "README.md"
 requires-python = ">=3.10"
 keywords = ["machine learning"]
diff --git a/uha/pytorch_oxe_dataloader.py b/uha/pytorch_oxe_dataloader.py
@@ -70,7 +70,7 @@ def __len__(self):
             ]
         )
         if hasattr(self._rlds_dataset, "sample_weights"):
-            lengths *= np.array(self._rlds_dataset.sample_weights)
+            lengths = np.array(self._rlds_dataset.sample_weights) * lengths
         total_len = lengths.sum()
         if self._is_train:
             return int(0.95 * total_len)

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ authors = [`
`9`	`9`	`{name = "Marcel Rühle", email="[email protected]"}`
`10`	`10`	`]`
`11`	`11`	`description = "Uha: Minimal adaption of the dataloader proposed in Octo for loading over 25 embodiments."`
`12`		`-version = "0.2.4"`
	`12`	`+version = "0.2.5"`
`13`	`13`	`readme = "README.md"`
`14`	`14`	`requires-python = ">=3.10"`
`15`	`15`	`keywords = ["machine learning"]`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ def __len__(self):`
`70`	`70`	`]`
`71`	`71`	`)`
`72`	`72`	`if hasattr(self._rlds_dataset, "sample_weights"):`
`73`		`- lengths *= np.array(self._rlds_dataset.sample_weights)`
	`73`	`+ lengths = np.array(self._rlds_dataset.sample_weights) * lengths`
`74`	`74`	`total_len = lengths.sum()`
`75`	`75`	`if self._is_train:`
`76`	`76`	`return int(0.95 * total_len)`