Merge pull request #403 from pcattori/hide-beta

Hide beta module from users
Datatamer · Jul 1, 2020 · cfb7d34 · cfb7d34
2 parents 33d0aa9 + a291f99
commit cfb7d34
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/tamr_client/__init__.py b/tamr_client/__init__.py
@@ -1,9 +1,9 @@
 # BETA check
 ############
 
-import tamr_client.beta as beta
+from tamr_client import _beta
 
-beta._check()
+_beta.check()
 
 # Logging
 #########
@@ -13,10 +13,13 @@
 # https://docs.python-guide.org/writing/logging/#logging-in-a-library
 logging.getLogger(__name__).addHandler(logging.NullHandler())
 
+# types
+#######
+
 from tamr_client._types import Attribute, AttributeType, SubAttribute, URL
 
-# Import shortcuts
-##################
+# functions
+###########
 
 # utilities
 from tamr_client import response

diff --git a/tamr_client/beta.py → tamr_client/_beta.py b/tamr_client/beta.py → tamr_client/_beta.py
@@ -2,7 +2,7 @@
 import sys
 
 
-def _check():
+def check():
     beta_flag = "TAMR_CLIENT_BETA"
     beta_enabled = "1"
     beta = os.environ.get(beta_flag)