Skip to content

Commit

Permalink
try to resolve OOM issue
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz committed Nov 7, 2024
1 parent 5e10621 commit fc7b6b7
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions source/tests/consistent/io/test_io.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
# SPDX-License-Identifier: LGPL-3.0-or-later
import copy
import gc
import shutil
import unittest
from pathlib import (
Expand Down Expand Up @@ -108,6 +109,9 @@ def test_data_equal(self):
data.pop(kk, None)
reference_data.pop(kk, None)
np.testing.assert_equal(data, reference_data)
# try to resolve OOM issue in the CI
del data, reference_data
gc.collect()

def test_deep_eval(self):
self.coords = np.array(
Expand Down Expand Up @@ -200,6 +204,8 @@ def test_deep_eval(self):
atomic=True,
)
rets_nopbc.append(ret)
del deep_eval
gc.collect()
for ret in rets[1:]:
for vv1, vv2 in zip(rets[0], ret):
if np.isnan(vv2).all():
Expand Down

0 comments on commit fc7b6b7

Please sign in to comment.