Skip to content

Commit 0332f07

Browse files
Merge pull request #28 from sightmachine/column_rename
Column name compatibility fix
2 parents 9f36203 + 113e99d commit 0332f07

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

smsdk/client.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -271,10 +271,12 @@ def get_machine_schema(self, machine_source, types=[], show_hidden=False, return
271271
if len(types) > 0:
272272
fields = [field for field in fields if field.get('type') in types]
273273

274+
frame = pd.DataFrame(fields).rename(columns={"display_name": "display", "type": "sight_type", "data_type": "type"})
275+
274276
if return_mtype:
275-
return (machine_type, pd.DataFrame(fields))
277+
return (machine_type, frame)
276278

277-
return pd.DataFrame(fields)
279+
return frame
278280

279281
def get_fields_of_machine_type(self, machine_type, types=[], show_hidden=False, **kwargs):
280282
machineType= smsdkentities.get('machine_type')

0 commit comments

Comments
 (0)