fix

Author: mgyucht

tests/databricks/sdk/service/jsonmarshallv2.py

@@ -11,8 +11,12 @@
 from google.protobuf.timestamp_pb2 import Timestamp
 
 from databricks.sdk.common.types.fieldmask import FieldMask
-from databricks.sdk.service._internal import (_duration, _enum, _from_dict,
-                                              _repeated_dict, _repeated_enum)
+from databricks.sdk.service._internal import (_duration, _enum, _fieldmask,
+                                              _from_dict, _repeated_dict,
+                                              _repeated_duration,
+                                              _repeated_enum,
+                                              _repeated_fieldmask,
+                                              _repeated_timestamp, _timestamp)
 
 _LOG = logging.getLogger("databricks.sdk")