Skip to content

Commit e9ae973

Browse files
committed
[TEST] Update CG solver test to enable GPU backend
Switched the `use_gpu` configuration to `True` in the CG solver test, ensuring the test runs with GPU acceleration. Removed unused imports for cleanup.
1 parent 4090e51 commit e9ae973

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

test/test_modules/test_cg/test_cg_solver.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,6 @@
11
import gempy as gp
22
from gempy.core.data.enumerators import ExampleModel
33
from gempy.optional_dependencies import require_gempy_viewer
4-
from gempy_engine.core.data.interp_output import InterpOutput
5-
6-
from test.verify_helper import gempy_verify_array
7-
import pytest
8-
from test.conftest import TEST_SPEED, TestSpeed
94

105
PLOT = True
116

@@ -18,7 +13,7 @@ def test_generate_greenstone_model():
1813
gempy_model=model,
1914
engine_config=gp.data.GemPyEngineConfig(
2015
backend=gp.data.AvailableBackends.PYTORCH,
21-
use_gpu=False,
16+
use_gpu=True,
2217
dtype='float32'
2318
)
2419
)

0 commit comments

Comments
 (0)