fixing imports for tf and torch in dataloader (#127)

* fixing imports for tf and torch in dataloader * fix tests imports of frameworks
NVIDIA-Merlin · Apr 5, 2023 · 699ca7b · 699ca7b
1 parent 4cd8f12
commit 699ca7b
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/merlin/dataloader/tensorflow.py b/merlin/dataloader/tensorflow.py
@@ -15,7 +15,7 @@
 #
 from functools import partial
 
-from merlin.core.compat import tensorflow as tf
+from merlin.core.compat.tensorflow import tensorflow as tf
 from merlin.dataloader.loader_base import LoaderBase
 from merlin.table import TensorColumn, TensorflowColumn, TensorTable
 from merlin.table.conversions import _dispatch_dlpack_fns, convert_col

diff --git a/merlin/dataloader/torch.py b/merlin/dataloader/torch.py
@@ -15,7 +15,7 @@
 #
 from functools import partial
 
-from merlin.core.compat import torch as th
+from merlin.core.compat.torch import torch as th
 from merlin.dataloader.loader_base import LoaderBase
 from merlin.table import TensorColumn, TensorTable, TorchColumn
 from merlin.table.conversions import _dispatch_dlpack_fns, convert_col

diff --git a/tests/unit/dataloader/test_array_to_tensorflow.py b/tests/unit/dataloader/test_array_to_tensorflow.py
@@ -16,7 +16,7 @@
 
 import pytest
 
-from merlin.core.compat import tensorflow as tf
+from merlin.core.compat.tensorflow import tensorflow as tf
 from merlin.core.dispatch import make_df
 from merlin.io import Dataset
 from merlin.schema import Tags

diff --git a/tests/unit/dataloader/test_array_to_torch.py b/tests/unit/dataloader/test_array_to_torch.py
@@ -15,7 +15,7 @@
 #
 import pytest
 
-from merlin.core.compat import torch as th
+from merlin.core.compat.torch import torch as th
 from merlin.core.dispatch import make_df
 from merlin.io import Dataset
 from merlin.schema import Tags