chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
import importlib
|
||||
import os
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
|
||||
@pytest.mark.offline
|
||||
class TestRDAgentImports(unittest.TestCase):
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
cls.rdagent_directory = Path(__file__).resolve().parent.parent.parent
|
||||
cls.modules = list(cls.import_all_modules_from_directory(cls.rdagent_directory))
|
||||
|
||||
@staticmethod
|
||||
def import_all_modules_from_directory(directory):
|
||||
for file in directory.joinpath("rdagent").rglob("*.py"):
|
||||
fstr = str(file)
|
||||
if "example" in fstr:
|
||||
continue
|
||||
if "meta_tpl" in fstr:
|
||||
continue
|
||||
if "autorl_bench/workspace/" in fstr:
|
||||
continue
|
||||
if "template" in fstr or "tpl" in fstr:
|
||||
continue
|
||||
if "model_coder" in fstr:
|
||||
continue
|
||||
if "llm_st" in fstr:
|
||||
continue
|
||||
if (
|
||||
"rdagent/log/ui/" in fstr
|
||||
or fstr.endswith("rdagent/app/cli.py")
|
||||
or fstr.endswith("rdagent/app/CI/run.py")
|
||||
or fstr.endswith("rdagent/app/utils/ape.py")
|
||||
or fstr.endswith("rdagent/log/ui/utils.py")
|
||||
):
|
||||
# the entrance points
|
||||
continue
|
||||
# llamafactory==0.9.3 pins numpy to an older version, causing other
|
||||
# installations to fail. The `extract_parameters` tests are therefore
|
||||
# temporarily disabled and can be re-enabled once the numpy constraint is relaxed.
|
||||
if "extract_parameters" in fstr:
|
||||
continue
|
||||
|
||||
yield fstr[fstr.index("rdagent") : -3].replace("/", ".")
|
||||
|
||||
def test_import_modules(self):
|
||||
print(self.modules)
|
||||
for module_name in self.modules:
|
||||
with self.subTest(module=module_name):
|
||||
try:
|
||||
print(module_name)
|
||||
importlib.import_module(module_name)
|
||||
except Exception as e:
|
||||
self.fail(f"Failed to import {module_name}: {e}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user