Update 3rd party libraries for ML

mwgamble · Jan 30, 2020 · d33c866 · d33c866
1 parent 1f21baa
commit d33c866
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/flake8_requirements/modules.py b/src/flake8_requirements/modules.py
@@ -506,7 +506,9 @@
     "pyopenssl": ["OpenSSL"],
     "python-dateutil": ["dateutil"],
     "pyyaml": ["yaml"],
+    "scikit-learn": ["sklearn"],
     "setuptools": ["pkg_resources", "setuptools"],
     "splunk_sdk": ["splunklib"],
+    "tensorboardx": ["tensorboardX"],
     "xlwt_future": ["xlwt"],
 }