strip the read file contents

stefanklut · Mar 26, 2024 · c57a8f0 · c57a8f0
1 parent 25ef761
commit c57a8f0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/core/setup.py b/core/setup.py
@@ -53,7 +53,7 @@ def get_git_hash() -> str:
 
     if version_path.is_file():
         with version_path.open(mode="r") as file:
-            git_hash = file.read()
+            git_hash = file.read().strip()
     else:
         git_hash = subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=Path(__file__).resolve().parent).strip().decode()
     return git_hash

diff --git a/run.py b/run.py
@@ -148,6 +148,7 @@ def cpu_call(self, data: AugInput) -> tuple[dict, int, int]:
 
             height, width, channels = data.image.shape
             assert channels == 3, f"Must be a RBG image, found {channels} channels"
+            # In place augmentation
             transform = self.aug(data)
             image = torch.as_tensor(data.image, dtype=torch.float32, device=self.cfg.MODEL.DEVICE).permute(2, 0, 1)