diff --git a/setup.py b/setup.py index 884f646..abb84bc 100644 --- a/setup.py +++ b/setup.py @@ -20,6 +20,7 @@ "torch", "pytorch_lightning", "pytorch-crf", + "hydra-configs-torch @ git+https://github.com/pytorch/hydra-torch/#subdirectory=hydra-configs-torch", ], test_requires=[ "pytest",