stage v0.1.7

tensordiffeq · Mar 27, 2021 · 9a1303a · 9a1303a
1 parent ba0e44c
commit 9a1303a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
 
 setuptools.setup(
     name="tensordiffeq",
-    version="0.1.6.7",
+    version="0.1.7",
     author="Levi McClenny",
     author_email="levimcclenny@tamu.edu",
     description="Distributed PDE Solver in Tensorflow",

diff --git a/tensordiffeq/output.py b/tensordiffeq/output.py
@@ -2,6 +2,8 @@
 from os import system, name
 import sys
 
-def print_screen(Domain, model):
+def print_screen(model):
     f = Figlet(font='slant')
-    print(f.renderText('TensorDiffEq'))
+    print(f.renderText('TensorDiffEq'))
+    print("Neural Network Model Summary\n")
+    print(model.u_model.summary())