Skip to content

Commit 6049b76

Browse files
committed
pass tags with **kwargs
1. add **kwargs to convert() in _converters_entry 2. remove extra space in test_api_examples.py
1 parent 250cf29 commit 6049b76

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

coremltools/converters/_converters_entry.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@ def convert(
6464
compute_units=_ComputeUnit.ALL,
6565
package_dir=None,
6666
debug=False,
67+
**kwargs
6768
):
6869
"""
6970
Convert a TensorFlow or PyTorch model to the Core ML model format as either
@@ -461,6 +462,7 @@ def skip_real_div_ops(op):
461462
package_dir=package_dir,
462463
debug=debug,
463464
specification_version=specification_version,
465+
**kwargs
464466
)
465467

466468
if exact_target == 'milinternal':

coremltools/test/api/test_api_examples.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,6 @@ def test_convert_from_saved_model_dir():
304304

305305
@staticmethod
306306
def test_convert_from_two_tags_saved_model_dir(tmpdir):
307-
308307
import tensorflow as tf
309308
from tensorflow.compat.v1.saved_model import build_tensor_info
310309
from tensorflow.compat.v1.saved_model import signature_constants

0 commit comments

Comments
 (0)