Update test_base.py

Author: Juntongkuki

tests/test_base.py

@@ -18,7 +18,7 @@
 import torch
 import os
 import sys
-from tnlearn.base import BaseModel
+from ..tnlearn.base import BaseModel
 
 
 # MockModel extends the base class BaseModel to provide a mock implementation.