tests: check for full paths in sha test

Haidra-Org · Apr 29, 2024 · 3b89c8a · 3b89c8a
1 parent 8a8ca6a
commit 3b89c8a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/model_managers/test_shared_model_manager.py b/tests/model_managers/test_shared_model_manager.py
@@ -69,7 +69,11 @@ def test_check_sha(
  continue
  if not (".pt" in path or ".ckpt" in path or ".safetensors" in path):
  continue
- model_manager.get_file_sha256_hash(f"{model_manager.model_folder_path}/{path}")
+ # Check if `path` is already a full path
+ if os.path.isabs(path):
+ model_manager.get_file_sha256_hash(path)
+ else:
+ model_manager.get_file_sha256_hash(f"{model_manager.model_folder_path}/{path}")
 
  def test_check_validate_all_available_models(
  self,