From c0401f4bcb1e1313b266d116c0b2b9faf57865eb Mon Sep 17 00:00:00 2001
From: njzjz-bot <njzjz.bot@gmail.com>
Date: Mon, 24 Aug 2026 04:33:25 +0800
Subject: [PATCH] Fix package imports in relax module

Use package-qualified imports for relax dependencies so dpti.relax works from an installed or repository-root context. Add an import regression test.

Coding-Agent: Codex
Codex-Version: codex-cli 0.149.0
Model: gpt-5.6-sol
Reasoning-Effort: xhigh
---
 dpti/relax.py       |  4 ++--
 tests/test_relax.py | 13 +++++++++++++
 2 files changed, 15 insertions(+), 2 deletions(-)
 create mode 100644 tests/test_relax.py

diff --git a/dpti/relax.py b/dpti/relax.py
index ec01c8bf..195d2793 100755
--- a/dpti/relax.py
+++ b/dpti/relax.py
@@ -4,8 +4,8 @@
 import json
 import os
 
-from lib.lammps import get_last_dump
-from lib.utils import create_path, cvt_conf
+from dpti.lib.lammps import get_last_dump
+from dpti.lib.utils import create_path, cvt_conf
 
 
 def _gen_lammps_relax(conf_file, mass_map, model, pres, thermo_freq=100, dump_freq=100):
diff --git a/tests/test_relax.py b/tests/test_relax.py
new file mode 100644
index 00000000..80241142
--- /dev/null
+++ b/tests/test_relax.py
@@ -0,0 +1,13 @@
+import unittest
+
+from dpti import relax
+
+
+class TestRelaxImports(unittest.TestCase):
+    def test_package_module_imports(self):
+        """The relax module exposes its public task generator when installed."""
+        self.assertTrue(callable(relax.make_task))
+
+
+if __name__ == "__main__":
+    unittest.main()