Add torchcodec in docstrings/tests for datasets 4.0 (#39156)
* fix dataset run_object_detection * bump version * keep same dataset actually * torchcodec in docstrings and testing utils * torchcodec in dockerfiles and requirements * remove duplicate * add torchocodec to all the remaining docker files * fix tests * support torchcodec in audio classification and ASR * [commit to revert] build ci-dev images * [commit to revert] trigger circleci * [commit to revert] build ci-dev images * fix * fix modeling_hubert * backward compatible run_object_detection * revert ci trigger commits * fix mono conversion and support torch tensor as input * revert map_to_array docs + fix it * revert mono * nit in docstring * style * fix modular --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -72,3 +72,14 @@ try:
|
||||
print("Number of TF GPUs available:", len(tf.config.list_physical_devices("GPU")))
|
||||
except ImportError:
|
||||
print("TensorFlow version:", None)
|
||||
|
||||
|
||||
try:
|
||||
import torchcodec
|
||||
|
||||
versions = torchcodec._core.get_ffmpeg_library_versions()
|
||||
print("FFmpeg version:", versions["ffmpeg_version"])
|
||||
except ImportError:
|
||||
print("FFmpeg version:", None)
|
||||
except (AttributeError, KeyError):
|
||||
print("Failed to get FFmpeg version")
|
||||
|
||||
Reference in New Issue
Block a user