fix optional dep. telekom#15

michalharakal · Aug 12, 2021 · d183718 · d183718
1 parent 7c8a28e
commit d183718
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -12,7 +12,7 @@
 project_name = "transformer_tools"
 source_code = "https://github.com/telekom/transformer-tools"
 keywords = "torch pytorch transformers nlp nlu"
-install_requires = ["torch", "transformers", "scikit-learn"]
+install_requires = ["torch", "transformers", "scikit-learn", "pandas", "numpy"]
 extras_require = {
     "checking": [
         "black",