Lora manager tests fix (#652)

This PR solves the "ModuleNotFoundError: No Module named torch.hpu" in test_lora_manager_hpu.py::test_from_lora_tensors by importing "habana_frameworks.hpu" to lora model. Co-authored-by: Vivek Goel <[email protected]>
HabanaAI · Jan 2, 2025 · 2012336 · 2012336
1 parent 2443ba9
commit 2012336
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/lora/test_lora_manager_hpu.py b/tests/lora/test_lora_manager_hpu.py
@@ -1,6 +1,7 @@
 import os
 from typing import Dict, List
 
+import habana_frameworks.torch  # noqa: F401
 import pytest
 import torch
 from safetensors.torch import load_file