Pretty print namespace

williamFalcon · Feb 10, 2018 · fa07f17 · fa07f17
1 parent 1a153d1
commit fa07f17
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 from setuptools import setup, find_packages
 
-version = '0.47'
+version = '0.48'
 
 setup(name='test_tube',
       packages=find_packages(),

diff --git a/test_tube/argparse_hopt.py b/test_tube/argparse_hopt.py
@@ -129,7 +129,7 @@ def __namespace_from_trial(self, trial):
             if k not in trial_dict:
                 trial_dict[k] = v
 
-        return argparse.Namespace(**trial_dict)
+        return TTNamespace(**trial_dict)
 
 
     def __flatten_params(self, params):
@@ -150,6 +150,14 @@ def __flatten_params(self, params):
         return flat_params
 
 
+class TTNamespace(argparse.Namespace):
+    def __str__(self):
+        result = '-'*100 + '\nHyperparameters:\n'
+        for k, v in self.__dict__.items():
+            result += '{0:20}: {1}\n'.format(k, v)
+        return result
+
+
 class OptArg(object):
 
     def __init__(self, obj_id, opt_values, nb_samples=None, tunnable=False):