(self, save_from_keras)
| 626 | |
| 627 | @parameterized.named_parameters(("SavedRaw", False), ("SavedFromKeras", True)) |
| 628 | def testGetConfigFromConfig(self, save_from_keras): |
| 629 | export_dir = os.path.join(self.get_temp_dir(), "half-plus-one") |
| 630 | _save_half_plus_one_model(export_dir, save_from_keras=save_from_keras) |
| 631 | layer = hub.KerasLayer(export_dir) |
| 632 | in_value = np.array([[10.0]], dtype=np.float32) |
| 633 | result = layer(in_value).numpy() |
| 634 | |
| 635 | config = layer.get_config() |
| 636 | new_layer = hub.KerasLayer.from_config(_json_cycle(config)) |
| 637 | new_result = new_layer(in_value).numpy() |
| 638 | self.assertEqual(result, new_result) |
| 639 | |
| 640 | def testGetConfigFromConfigWithHParams(self): |
| 641 | if tf.__version__ == "2.0.0-alpha0": |
nothing calls this directly
no test coverage detected