From f52406703c78273fa2bf89c545069e0da999c220 Mon Sep 17 00:00:00 2001 From: Gao Enhao Date: Fri, 8 Dec 2023 23:17:49 +0800 Subject: [PATCH] [FIX] fix flake8 error --- tests/conftest.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 89d73cb..d2510fc 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -2,15 +2,11 @@ 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 from abl.structures import ListData -from examples.models.nn import LeNet5 +from ..examples.models.nn import LeNet5 # Fixture for BasicNN instance