From 332f6eeee12de6792e207e4ea990201346ae6c16 Mon Sep 17 00:00:00 2001 From: Gao Enhao Date: Fri, 8 Dec 2023 23:08:11 +0800 Subject: [PATCH] [FIX] add path in conftest --- tests/conftest.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/conftest.py b/tests/conftest.py index 09a8247..89d73cb 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -2,6 +2,10 @@ import pytest import torch import torch.nn as nn import torch.optim as optim +import os +import sys + +sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from abl.learning import BasicNN from abl.reasoning import GroundKB, KBBase, PrologKB, Reasoner