Fix CI to include parameterized for testing (#247)

Summary: Pull Request resolved: #247 9b69c94 adds a dependency on this for testing that wasnt present in dev-requirements Reviewed By: daniellepintz Differential Revision: D40539766 fbshipit-source-id: c8f974c2d7e183e0b10fce700ef27786dca6e7a0
pytorch · Oct 20, 2022 · bf36265 · bf36265
1 parent 9d9a237
commit bf36265
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -1,2 +1,3 @@
+parameterized
 pytest
 pytest-cov
diff --git a/torchtnt/runner/train.py b/torchtnt/runner/train.py
@@ -5,7 +5,7 @@
 # LICENSE file in the root directory of this source tree.
 
 import logging
-from typing import cast, Iterable, List, Optional
+from typing import Iterable, List, Optional
 
 import torch
 from torchtnt.runner.callback import Callback