Skip to content

Commit abc0289

Browse files
committed
make use external sumkdft_workers
1 parent 50eb627 commit abc0289

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

src/dcore/sumkdft_workers/mpi_main.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,20 @@
1414
args = parser.parse_args()
1515

1616
# Runner class
17-
m = importlib.import_module('dcore.sumkdft_workers.all_workers')
18-
cls = getattr(m, args.runner_cls)
17+
m_cls = args.runner_cls.split('.')
18+
if len(m_cls) == 1:
19+
m = importlib.import_module('dcore.sumkdft_workers.all_workers')
20+
cls = getattr(m, args.runner_cls)
21+
else:
22+
m = importlib.import_module('.'.join(m_cls[:-1]))
23+
cls = getattr(m, m_cls[-1])
1924
runner = cls(args.model_hdf5_file, args.input_file, args.output_file)
2025
runner.run()
21-
26+
except ImportError as e:
27+
print("Import error:", e)
28+
import traceback
29+
traceback.print_exc()
30+
sys.exit(1)
2231

2332
except Exception as e:
2433
print("Unexpected error:", e)

0 commit comments

Comments
 (0)