Skip to content

Commit 436e741

Browse files
committed
refactor: update config access from dict to attribute-style in test case
1 parent 4dfd43f commit 436e741

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

extension/llm/export/test/test_export_llm.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -142,13 +142,13 @@ def test_with_config_and_cli(self, mock_export_llama: MagicMock) -> None:
142142
mock_export_llama.assert_called_once()
143143
called_config = mock_export_llama.call_args[0][0]
144144
self.assertEqual(
145-
called_config["base"]["model_class"], "stories110m"
145+
called_config.base.model_class, "stories110m"
146146
) # Override from CLI.
147147
self.assertEqual(
148-
called_config["model"]["dtype_override"].value, "fp16"
148+
called_config.model.dtype_override.value, "fp16"
149149
) # From yaml.
150150
self.assertEqual(
151-
called_config["backend"]["xnnpack"]["enabled"],
151+
called_config.backend.xnnpack.enabled,
152152
True, # Override from CLI.
153153
)
154154
finally:

0 commit comments

Comments
 (0)