File tree Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Original file line number Diff line number Diff 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' :
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments