Skip to content
This repository was archived by the owner on Aug 25, 2024. It is now read-only.

Commit bb81494

Browse files
author
John Andersen
committed
model: xgboost: Temporary directory for tests
Signed-off-by: John Andersen <[email protected]>
1 parent 2c0aeb9 commit bb81494

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

model/xgboost/tests/test_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from dffml.model.accuracy import Accuracy
1313
from dffml import train, accuracy, predict
1414
from dffml.util.entrypoint import entrypoint
15-
from dffml.util.asynctestcase import AsyncTestCase
15+
from dffml.util.asynctestcase import IntegrationCLITestCase
1616
from dffml.feature.feature import Feature, Features
1717
from dffml.model.model import SimpleModel, ModelNotTrained
1818
from dffml.source.memory import MemorySource, MemorySourceConfig
@@ -24,7 +24,7 @@
2424
)
2525

2626

27-
class TestXGBRegressor(AsyncTestCase):
27+
class TestXGBRegressor(IntegrationCLITestCase):
2828
@classmethod
2929
def setUpClass(cls):
3030
# Create a temporary directory to store the trained model

0 commit comments

Comments
 (0)