small fixes

FAIR-Chem · Sep 20, 2024 · de70c2f · de70c2f
1 parent 89dbc7d
commit de70c2f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/src/fairchem/core/_cli_hydra.py b/src/fairchem/core/_cli_hydra.py
@@ -104,7 +104,7 @@ def main(
         )
         job = executor.submit(runner_wrapper, cfg, args)
         logger.info(
-            f"Submitted job id: {timestamp_id}, slurm id: {job}, logs: {log_dir}"
+            f"Submitted job id: {timestamp_id}, slurm id: {job.job_id}, logs: {log_dir}"
         )
     else:
         if args.num_gpus > 1:

diff --git a/src/fairchem/core/common/distutils.py b/src/fairchem/core/common/distutils.py
@@ -220,9 +220,6 @@ def gather_objects(data: T, group: dist.ProcessGroup = dist.group.WORLD) -> list
 
 
 def assign_device_for_local_rank(cpu: bool, local_rank: int):
-    assert (
-        os.environ.get(CURRENT_DEVICE_STR) is None
-    ), "environment variable CURRENT_DEVICE is used for another purpose!"
     if cpu:
         os.environ[CURRENT_DEVICE_STR] = "cpu"
     else: