summary fix

wandb/apis/internal.py

@@ -70,6 +70,9 @@ def set_setting(self, *args, **kwargs):
     def parse_slug(self, *args, **kwargs):
         return self.api.parse_slug(*args, **kwargs)
 
+    def download_url(self, *args, **kwargs):
+        return self.api.download_url(*args, **kwargs)
+
     def download_urls(self, *args, **kwargs):
         return self.api.download_urls(*args, **kwargs)
 

wandb/old/summary.py

@@ -26,7 +26,6 @@
 
 DEEP_SUMMARY_FNAME = 'wandb.h5'
 H5_TYPES = ("numpy.ndarray", "tensorflow.Tensor", "torch.Tensor")
-
 h5py = util.get_module("h5py")
 np = util.get_module("numpy")
 
@@ -38,13 +37,17 @@ class SummarySubDict(object):
     """
 
     def __init__(self, root=None, path=()):
+        self._path = tuple(path)
         if root is None:
             self._root = self
+            self._json_dict = {}
         else:
             self._root = root
-        self._path = tuple(path)
+            json_dict = root._json_dict
+            for k in path:
+                json_dict = json_dict[k]
+            self._json_dict = json_dict
         self._dict = {}
-        self._json_dict = {}
 
         # We use this to track which keys the user has set explicitly
         # so that we don't automatically overwrite them when we update