Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
[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.
  • Loading branch information
Leguark committed Sep 25, 2025
commit 7627a43b6ccdf7ac133777be273faa65ac3f93e3
7 changes: 1 addition & 6 deletions test/test_modules/test_cg/test_cg_solver.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,6 @@
import gempy as gp
from gempy.core.data.enumerators import ExampleModel
from gempy.optional_dependencies import require_gempy_viewer
from gempy_engine.core.data.interp_output import InterpOutput

from test.verify_helper import gempy_verify_array
import pytest
from test.conftest import TEST_SPEED, TestSpeed

PLOT = True

Expand All @@ -18,7 +13,7 @@ def test_generate_greenstone_model():
gempy_model=model,
engine_config=gp.data.GemPyEngineConfig(
backend=gp.data.AvailableBackends.PYTORCH,
use_gpu=False,
use_gpu=True,
dtype='float32'
)
)
Expand Down