Esempio n. 1
0
    def test_set_quantization_sampling_size_on_empty_config(self) -> None:
        """Test set_model_path."""
        config = Config()

        config.set_quantization_sampling_size("new sampling size")

        self.assertIsNone(config.quantization)
Esempio n. 2
0
    def test_set_quantization_sampling_size(self) -> None:
        """Test set_model_path."""
        config = Config(self.predefined_config)

        config.set_quantization_sampling_size("new sampling size")

        self.assertEqual("new sampling size",
                         config.quantization.calibration.sampling_size)