Skip to content

Commit 68dbd0f

Browse files
committed
MAINT replace oml: by oml-python: to avoid confusion with openml-defined objects/entities
1 parent fbf05de commit 68dbd0f

2 files changed

Lines changed: 15 additions & 13 deletions

File tree

openml/flows/sklearn_converter.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -96,8 +96,8 @@ def flow_to_sklearn(o, **kwargs):
9696
# Check if the dict encodes a 'special' object, which could not
9797
# easily converted into a string, but rather the information to
9898
# re-create the object were stored in a dictionary.
99-
elif 'oml:serialized_object' in o:
100-
serialized_type = o['oml:serialized_object']
99+
elif 'oml-python:serialized_object' in o:
100+
serialized_type = o['oml-python:serialized_object']
101101
value = o['value']
102102
if serialized_type == 'type':
103103
rval = deserialize_type(value, **kwargs)
@@ -202,7 +202,7 @@ def _serialize_model(model):
202202
sub_component_identifier = k + '__' + identifier
203203
sub_components[sub_component_identifier] = sub_component
204204
component_reference = OrderedDict()
205-
component_reference['oml:serialized_object'] = 'component_reference'
205+
component_reference['oml-python:serialized_object'] = 'component_reference'
206206
component_reference['value'] = OrderedDict(
207207
key=sub_component_identifier, step_name=identifier)
208208
parameter_value.append(component_reference)
@@ -224,7 +224,7 @@ def _serialize_model(model):
224224
sub_components[k] = rval
225225
sub_components_explicit.add(k)
226226
component_reference = OrderedDict()
227-
component_reference['oml:serialized_object'] = 'component_reference'
227+
component_reference['oml-python:serialized_object'] = 'component_reference'
228228
component_reference['value'] = OrderedDict(key=k, step_name=None)
229229
component_reference = sklearn_to_flow(component_reference)
230230
parameters[k] = json.dumps(component_reference)
@@ -306,7 +306,7 @@ def _deserialize_model(flow, **kwargs):
306306
rval = flow_to_sklearn(value, components=components)
307307

308308
# Replace the component placeholder by the actual flow
309-
if isinstance(rval, dict) and 'oml:serialized_object' in rval:
309+
if isinstance(rval, dict) and 'oml-python:serialized_object' in rval:
310310
parameter_name, step = rval['value'].split('__')
311311
if parameter_name not in component_dict:
312312
component_dict[parameter_name] = OrderedDict()
@@ -334,7 +334,7 @@ def serialize_type(o):
334334
np.int32: 'np.int32',
335335
np.int64: 'np.int64'}
336336
ret = OrderedDict()
337-
ret['oml:serialized_object'] = 'type'
337+
ret['oml-python:serialized_object'] = 'type'
338338
ret['value'] = mapping[o]
339339
return ret
340340

@@ -358,7 +358,7 @@ def serialize_rv_frozen(o):
358358
b = o.b
359359
dist = o.dist.__class__.__module__ + '.' + o.dist.__class__.__name__
360360
ret = OrderedDict()
361-
ret['oml:serialized_object'] = 'rv_frozen'
361+
ret['oml-python:serialized_object'] = 'rv_frozen'
362362
ret['value'] = OrderedDict(dist=dist, a=a, b=b, args=args, kwds=kwds)
363363
return ret
364364

@@ -387,7 +387,7 @@ def deserialize_rv_frozen(o, **kwargs):
387387
def serialize_function(o):
388388
name = o.__module__ + '.' + o.__name__
389389
ret = OrderedDict()
390-
ret['oml:serialized_object'] = 'function'
390+
ret['oml-python:serialized_object'] = 'function'
391391
ret['value'] = name
392392
return ret
393393

@@ -440,7 +440,7 @@ def _serialize_cross_validator(o):
440440
else:
441441
parameters[key] = None
442442

443-
ret['oml:serialized_object'] = 'cv_object'
443+
ret['oml-python:serialized_object'] = 'cv_object'
444444
name = o.__module__ + "." + o.__class__.__name__
445445
value = OrderedDict(name=name, parameters=parameters)
446446
ret['value'] = value

tests/flows/test_sklearn.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -127,8 +127,10 @@ def test_serialize_pipeline(self):
127127
# Hard to compare two representations of a dict due to possibly
128128
# different sorting. Making a json makes it easier
129129
self.assertEqual(json.loads(serialization.parameters['steps']),
130-
[{'oml:serialized_object': 'component_reference', 'value': {'key': 'steps__scaler', 'step_name': 'scaler'}},
131-
{'oml:serialized_object': 'component_reference', 'value': {'key': 'steps__dummy', 'step_name': 'dummy'}}])
130+
[{'oml-python:serialized_object':
131+
'component_reference', 'value': {'key': 'steps__scaler', 'step_name': 'scaler'}},
132+
{'oml-python:serialized_object':
133+
'component_reference', 'value': {'key': 'steps__dummy', 'step_name': 'dummy'}}])
132134

133135
# Checking the sub-component
134136
self.assertEqual(len(serialization.components), 2)
@@ -273,12 +275,12 @@ def test_serialize_function(self):
273275
def test_serialize_cvobject(self):
274276
methods = [sklearn.model_selection.KFold(3),
275277
sklearn.model_selection.LeaveOneOut()]
276-
fixtures = [OrderedDict([('oml:serialized_object', 'cv_object'),
278+
fixtures = [OrderedDict([('oml-python:serialized_object', 'cv_object'),
277279
('value', OrderedDict([('name', 'sklearn.model_selection._split.KFold'),
278280
('parameters', OrderedDict([('n_splits', '3'),
279281
('random_state', 'null'),
280282
('shuffle', 'false')]))]))]),
281-
OrderedDict([('oml:serialized_object', 'cv_object'),
283+
OrderedDict([('oml-python:serialized_object', 'cv_object'),
282284
('value', OrderedDict([('name', 'sklearn.model_selection._split.LeaveOneOut'),
283285
('parameters', OrderedDict())]))])]
284286
for method, fixture in zip(methods, fixtures):

0 commit comments

Comments
 (0)