refactored importing

benranderson · Jun 9, 2018 · 0173217 · 0173217
1 parent 9eb7b5a
commit 0173217
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/wallthick/__init__.py b/wallthick/__init__.py
@@ -8,6 +8,6 @@
 
 __author__ = """Ben Randerson"""
 __email__ = 'ben.m.randerson@email.com'
-__version__ = '0.0.4'
+__version__ = '0.0.6'
 
-from .pd8010 import *
+from .pd8010 import Pd8010
diff --git a/wallthick/cli.py b/wallthick/cli.py
@@ -6,7 +6,7 @@
 import click
 import json
 
-from wallthick.pd8010 import Pd8010
+import wallthick
 
 req_inputs = [
     't_sel',
@@ -38,7 +38,7 @@ def main(inputs):
     if all(param in data for param in req_inputs):
         click.secho(
             'Running PD 8010-2 wall thickness calculation...', fg='green')
-        pd = Pd8010(data)
+        pd = wallthick.Pd8010(data)
         click.echo(f'\nNominal Wall Thicknesses')
         click.echo(f'------------------------')
         click.echo(f'Pressure Containment:\t{1000*pd.t_h:.3f} mm')