| import importlib | |
| import unittest | |
| class ExampleImportTests(unittest.TestCase): | |
| def test_example_scripts_import_without_ml_dependencies(self): | |
| infer = importlib.import_module("examples.infer_transformers") | |
| endpoint = importlib.import_module("examples.call_local_endpoint") | |
| server = importlib.import_module("examples.serve_local_endpoint") | |
| self.assertEqual(infer.DEFAULT_BASE_MODEL, "Qwen/Qwen2.5-3B-Instruct") | |
| self.assertEqual(server.DEFAULT_ADAPTER_PATH, ".") | |
| payload = endpoint.build_payload("Explain delayed neutrons.", model="local", max_tokens=128) | |
| self.assertEqual(payload["messages"][1]["role"], "user") | |
| self.assertIn("civil-nuclear", payload["messages"][0]["content"]) | |
| if __name__ == "__main__": | |
| unittest.main() | |