Spaces:
Running
on
Zero
Running
on
Zero
Fix SparseGAState class
Browse files- 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.
|
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():
|