Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Summary: previous diff D53456248 mistakenly pinned the torchvision version in the mnist example. We want to pin the version in the dev-requirements instead to fix the following error: ``` ==================================== ERRORS ==================================== __________________ ERROR collecting tests/utils/test_flops.py __________________ tests/utils/test_flops.py:11: in <module> import torchvision.models as models /usr/share/miniconda3/envs/test/lib/python3.9/site-packages/torchvision/__init__.py:6: in <module> from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils /usr/share/miniconda3/envs/test/lib/python3.9/site-packages/torchvision/_meta_registrations.py:164: in <module> def meta_nms(dets, scores, iou_threshold): /usr/share/miniconda3/envs/test/lib/python3.9/site-packages/torch/library.py:467: in inner handle = entry.abstract_impl.register(func_to_register, source) /usr/share/miniconda3/envs/test/lib/python3.9/site-packages/torch/_library/abstract_impl.py:30: in register if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"): E RuntimeError: operator torchvision::nms does not exist _____________ ERROR collecting tests/utils/test_module_summary.py ______________ tests/utils/test_module_summary.py:18: in <module> from torchvision import models /usr/share/miniconda3/envs/test/lib/python3.9/site-packages/torchvision/__init__.py:6: in <module> from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils /usr/share/miniconda3/envs/test/lib/python3.9/site-packages/torchvision/_meta_registrations.py:26: in <module> def meta_roi_align(input, rois, spatial_scale, pooled_height, pooled_width, sampling_ratio, aligned): /usr/share/miniconda3/envs/test/lib/python3.9/site-packages/torchvision/_meta_registrations.py:18: in wrapper if torchvision.extension._has_ops(): E AttributeError: partially initialized module 'torchvision' has no attribute 'extension' (most likely due to a circular import) =============================== warnings summary =============================== torchtnt/utils/version.py:10 /home/runner/work/tnt/tnt/torchtnt/utils/version.py:10: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html import pkg_resources tests/framework/test_unit.py:18 /home/runner/work/tnt/tnt/tests/framework/test_unit.py:18: PytestCollectionWarning: cannot collect test class 'TestUnit' because it has a __init__ constructor (from: tests/framework/test_unit.py) class TestUnit( -- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html ---------- coverage: platform linux, python 3.9.18-final-0 ----------- Coverage XML written to file coverage.xml =========================== short test summary info ============================ ERROR tests/utils/test_flops.py - RuntimeError: operator torchvision::nms does not exist ERROR tests/utils/test_module_summary.py - AttributeError: partially initialized module 'torchvision' has no attribute 'extension' (most likely due to a circular import) !!!!!!!!!!!!!!!!!!! Interrupted: 2 errors during collection !!!!!!!!!!!!!!!!!!!! ======================== 2 warnings, 2 errors in 3.40s ========================= Error: Process completed with exit code 2. ``` Reviewed By: gunchu Differential Revision: D53720066 fbshipit-source-id: d9f8a333e539e5492027159e0ab434db1f663ed1
- Loading branch information