Merge pull request #36 from nicklashansen/main

refactor orca->octo in pytorch example
octo-models · Jan 19, 2024 · 86df8bd · 86df8bd
2 parents 8fe7497 + b4ea6fe
commit 86df8bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/06_pytorch_oxe_dataloader.py b/examples/06_pytorch_oxe_dataloader.py
@@ -4,8 +4,8 @@
 shuffle buffer size, which are reduced for demonstration purposes).
 """
 import numpy as np
-from orca.data.dataset import make_interleaved_dataset
-from orca.data.oxe import make_oxe_dataset_kwargs_and_weights
+from octo.data.dataset import make_interleaved_dataset
+from octo.data.oxe import make_oxe_dataset_kwargs_and_weights
 import tensorflow as tf
 import torch
 from torch.utils.data import DataLoader