def __init__(self, cache_dir=None): """Creates a resolver that streams tar/gz file content over HTTP. Args: cache_dir: directory to download and cache modules to. """ self._cache_dir = resolver.tfhub_cache_dir(cache_dir, use_temp=True)
def testCacheDir(self): # No cache dir set, None is returned. cache_dir = resolver.tfhub_cache_dir() self.assertEqual(cache_dir, None) # Use temp dir. cache_dir = resolver.tfhub_cache_dir(use_temp=True) self.assertEquals(cache_dir, os.path.join(tempfile.gettempdir(), "tfhub_modules")) # Use override cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True) self.assertEqual("/d", cache_dir) # Use a flag FLAGS.tfhub_cache_dir = "/e" cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True) self.assertEqual("/e", cache_dir) FLAGS.tfhub_cache_dir = "" # Use env variable os.environ[resolver._TFHUB_CACHE_DIR] = "/f" cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True) self.assertEqual("/f", cache_dir) FLAGS.tfhub_cache_dir = "/e" cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True) self.assertEqual("/f", cache_dir) FLAGS.tfhub_cache_dir = "" os.unsetenv(resolver._TFHUB_CACHE_DIR)
def testCacheDir(self): # No cache dir set, None is returned. cache_dir = resolver.tfhub_cache_dir() self.assertEqual(cache_dir, None) # Use temp dir. cache_dir = resolver.tfhub_cache_dir(use_temp=True) self.assertEquals(cache_dir, os.path.join(tempfile.gettempdir(), "tfhub_modules")) # Use override cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True) self.assertEqual("/d", cache_dir) # Use a flag FLAGS.tfhub_cache_dir = "/e" cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True) self.assertEqual("/e", cache_dir) FLAGS.tfhub_cache_dir = "" # Use env variable os.environ[resolver._TFHUB_CACHE_DIR] = "/f" cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True) self.assertEqual("/f", cache_dir) FLAGS.tfhub_cache_dir = "/e" cache_dir = resolver.tfhub_cache_dir(default_cache_dir="/d", use_temp=True) self.assertEqual("/f", cache_dir) FLAGS.tfhub_cache_dir = "" os.unsetenv(resolver._TFHUB_CACHE_DIR)
def _module_dir(handle): """Returns the directory where to cache the module.""" cache_dir = resolver.tfhub_cache_dir(use_temp=True) return resolver.create_local_module_dir( cache_dir, hashlib.sha1(handle.encode("utf8")).hexdigest())