Keep samplers private - for now

pytorch · Jun 20, 2024 · b6d0b80 · b6d0b80
1 parent d4b0111
commit b6d0b80
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 4 deletions.
diff --git a/src/torchcodec/samplers/__init__.py b/src/torchcodec/samplers/__init__.py
@@ -1,5 +1,2 @@
 # TODO: We need to figure out what we want to be public and what we want to be
 # private.
-# TODO: Don't use import *
-
-from .video_clip_sampler import *  # noqa
diff --git a/...torchcodec/samplers/video_clip_sampler.py → ...orchcodec/samplers/_video_clip_sampler.py b/...torchcodec/samplers/video_clip_sampler.py → ...orchcodec/samplers/_video_clip_sampler.py
diff --git a/test/samplers/video_clip_sampler_test.py b/test/samplers/video_clip_sampler_test.py
@@ -3,7 +3,7 @@
 
 import pytest
 import torch
-from torchcodec.samplers import (
+from torchcodec.samplers._video_clip_sampler import (
     IndexBasedSamplerArgs,
     TimeBasedSamplerArgs,
     VideoArgs,