easyvolcap: import fix bugs

zju3dv · Apr 6, 2024 · 86e5f2b · 86e5f2b
1 parent 4fc98a1
commit 86e5f2b
Show file tree

Hide file tree

Showing 12 changed files with 24 additions and 12 deletions.
diff --git a/easyvolcap/dataloaders/__init__.py b/easyvolcap/dataloaders/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/dataloaders/datasets/__init__.py b/easyvolcap/dataloaders/datasets/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/models/__init__.py b/easyvolcap/models/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/models/cameras/__init__.py b/easyvolcap/models/cameras/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/models/networks/__init__.py b/easyvolcap/models/networks/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/models/networks/embedders/__init__.py b/easyvolcap/models/networks/embedders/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/models/renderers/__init__.py b/easyvolcap/models/renderers/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/models/samplers/__init__.py b/easyvolcap/models/samplers/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/models/supervisors/__init__.py b/easyvolcap/models/supervisors/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/runners/__init__.py b/easyvolcap/runners/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/runners/evaluators/__init__.py b/easyvolcap/runners/evaluators/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))
diff --git a/easyvolcap/runners/visualizers/__init__.py b/easyvolcap/runners/visualizers/__init__.py
@@ -6,11 +6,12 @@
         # from . import *  # the actual imports
         exec(f'from . import {module}')
     except Exception as e:
+        import os
         import sys
         from easyvolcap.utils.console_utils import *
 
         exception_type, exception_object, exception_traceback = sys.exc_info()
         filename = exception_traceback.tb_frame.f_code.co_filename
         line_number = exception_traceback.tb_lineno
 
-        log(yellow(f'Failed to import {red(filename)}:{line_number}, {red(type(e))}: {red_slim(e)}'))
+        log(yellow(f'Failed to import {red(basename(filename))}:{red(line_number)}, {red(type(e).__name__)}: {e}'))