|
8 | 8 | from importlib import import_module |
9 | 9 |
|
10 | 10 | CONFIG_LOCATIONS = { |
11 | | - "LazyTensorCoreTestConfig" : "lazy_tensor_core", |
12 | | - "NativeTorchTestConfig" : "native_torch", |
13 | | - "OnnxBackendTestConfig" : "onnx_backend", |
14 | | - "TorchScriptTestConfig" : "torchscript", |
15 | | - "TorchDynamoTestConfig" : "torchdynamo", |
16 | | - "JITImporterTestConfig" : "jit_importer_backend", |
17 | | - "FxImporterTestConfig" : "fx_importer_backend", |
| 11 | + "LazyTensorCoreTestConfig": "lazy_tensor_core", |
| 12 | + "NativeTorchTestConfig": "native_torch", |
| 13 | + "OnnxBackendTestConfig": "onnx_backend", |
| 14 | + "TorchScriptTestConfig": "torchscript", |
| 15 | + "TorchDynamoTestConfig": "torchdynamo", |
| 16 | + "JITImporterTestConfig": "jit_importer_backend", |
| 17 | + "FxImporterTestConfig": "fx_importer_backend", |
18 | 18 | } |
19 | 19 |
|
20 | 20 | def load_config(name: str) -> type: |
21 | 21 | source = CONFIG_LOCATIONS.get(name) |
22 | 22 | assert source is not None, f"Could not find TestConfig named {name}." |
23 | | - module = import_module(f'.{source}', __package__) |
| 23 | + module = import_module(f".{source}", __package__) |
24 | 24 | return getattr(module, name) |
0 commit comments