Ejemplos de ModelCheckpointMgr._getModelDir en Python

Ejemplo n.º 1

Archivo: model_checkpoint_mgr_test.py Proyecto: sergius/numenta-apps

  def testStorageDirOverrideViaEnvironmentVariable(self):
    with ModelCheckpointStoragePatch() as storagePatch:
      checkpointMgr = ModelCheckpointMgr()

      tempModelCheckpointDir = storagePatch.tempModelCheckpointDir
      modelEntryDir = checkpointMgr._getModelDir(modelID="abc", mustExist=False)

    self.assertIn(tempModelCheckpointDir, modelEntryDir)

Ejemplo n.º 2

Archivo: model_checkpoint_mgr_test.py Proyecto: ysei/numenta-apps-archived-htm.it-grokcli-htmitmobile

  def testStorageDirOverrideViaEnvironmentVariable(self):
    with ModelCheckpointStoragePatch() as storagePatch:
      checkpointMgr = ModelCheckpointMgr()

      tempModelCheckpointDir = storagePatch.tempModelCheckpointDir
      modelEntryDir = checkpointMgr._getModelDir(modelID="abc", mustExist=False)

    self.assertIn(tempModelCheckpointDir, modelEntryDir)