Added auto sub module

LucaCappelletti94 · Mar 9, 2020 · 440f6a6 · 440f6a6
1 parent c127b5c
commit 440f6a6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/silence_tensorflow/auto.py b/silence_tensorflow/auto.py
@@ -0,0 +1,3 @@
+from .silence_tensorflow import silence_tensorflow
+
+silence_tensorflow()
diff --git a/tests/test_silence_tensorflow.py b/tests/test_silence_tensorflow.py
@@ -2,4 +2,7 @@
 
 def test_silence_tensorflow():
     """Check that everything runs."""
-    silence_tensorflow()
+    silence_tensorflow()
+
+def test_auto_silence():
+    import silence_tensorflow.auto