ostapagon commited on
Commit
014e82e
·
1 Parent(s): da0f211

Fix SparseGAState class

Browse files
Files changed (1) hide show
  1. demo/mast3r_demo.py +2 -4
demo/mast3r_demo.py CHANGED
@@ -43,12 +43,10 @@ class SparseGAState():
43
  self.outfile_name = outfile_name
44
 
45
  def __del__(self):
46
- if self.cache_dir is not None and os.path.isdir(self.cache_dir):
47
  shutil.rmtree(self.cache_dir)
48
- self.cache_dir = None
49
- if self.outfile_name is not None and os.path.isfile(self.outfile_name):
50
  os.remove(self.outfile_name)
51
- self.outfile_name = None
52
 
53
 
54
  def get_args_parser():
 
43
  self.outfile_name = outfile_name
44
 
45
  def __del__(self):
46
+ if hasattr(self, 'cache_dir') and self.cache_dir is not None and os.path.isdir(self.cache_dir):
47
  shutil.rmtree(self.cache_dir)
48
+ if hasattr(self, 'outfile_name') and self.outfile_name is not None and os.path.isfile(self.outfile_name):
 
49
  os.remove(self.outfile_name)
 
50
 
51
 
52
  def get_args_parser():