File tree Expand file tree Collapse file tree 3 files changed +4
-16
lines changed
merlin/loader/ops/embeddings Expand file tree Collapse file tree 3 files changed +4
-16
lines changed Original file line number Diff line number Diff line change 1313# See the License for the specific language governing permissions and
1414# limitations under the License.
1515#
16-
17- # flake8: noqa
18- from merlin .loader .ops .embeddings .tf_embedding_op import (
19- TF_MmapNumpyTorchEmbedding ,
20- TF_NumpyEmbeddingOperator ,
21- TFEmbeddingOperator ,
22- )
23- from merlin .loader .ops .embeddings .torch_embedding_op import (
24- Torch_MmapNumpyTorchEmbedding ,
25- Torch_NumpyEmbeddingOperator ,
26- TorchEmbeddingOperator ,
27- )
Original file line number Diff line number Diff line change 2020
2121from merlin .core .dispatch import HAS_GPU
2222from merlin .io import Dataset
23- from merlin .loader .tensorflow import Loader
2423from merlin .schema import Tags
2524
2625tf = pytest .importorskip ("tensorflow" )
2726
28- from merlin .loader .ops .embeddings import ( # noqa
27+ from merlin .loader .ops .embeddings . tf_embedding_op import ( # noqa
2928 TF_MmapNumpyTorchEmbedding ,
3029 TF_NumpyEmbeddingOperator ,
3130 TFEmbeddingOperator ,
3231)
32+ from merlin .loader .tensorflow import Loader # noqa
3333
3434
3535@pytest .mark .parametrize ("cpu" , [None , "cpu" ] if HAS_GPU else ["cpu" ])
Original file line number Diff line number Diff line change 2020
2121from merlin .core .dispatch import HAS_GPU
2222from merlin .io import Dataset
23- from merlin .loader .torch import Loader
2423from merlin .schema import Tags
2524
2625torch = pytest .importorskip ("torch" )
2726
28- from merlin .loader .ops .embeddings import ( # noqa
27+ from merlin .loader .ops .embeddings . torch_embedding_op import ( # noqa
2928 Torch_MmapNumpyTorchEmbedding ,
3029 Torch_NumpyEmbeddingOperator ,
3130 TorchEmbeddingOperator ,
3231)
32+ from merlin .loader .torch import Loader # noqa
3333
3434
3535@pytest .mark .parametrize ("cpu" , [None , "cpu" ] if HAS_GPU else ["cpu" ])
You can’t perform that action at this time.
0 commit comments