Skip to content

Commit 655f8f2

Browse files
committed
Auto add Machine and End Time if not present
1 parent 8170df5 commit 655f8f2

2 files changed

Lines changed: 8 additions & 3 deletions

File tree

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515

1616
setup(
1717
name="smsdk",
18-
version="1.0.16",
18+
version="1.0.17",
1919
packages=find_packages(exclude=["test*"]),
2020
include_package_data=True,
2121
install_requires=install_requires,

smsdk/client_v0.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -295,8 +295,13 @@ def inner(self, normalize=True, clean_strings_in=True, clean_strings_out=True, *
295295
kwargs['_only'] = only_names + toplevel
296296
else:
297297
if ('Machine' not in kwargs['_only']) and ('machine__source' not in kwargs['_only']):
298-
print("please provide Machine / machine__source in _only field")
299-
return pd.DataFrame()
298+
print("Adding Machine to _only")
299+
kwargs['_only'].append('Machine')
300+
301+
if ('End Time' not in kwargs['_only']) and ('endtime' not in kwargs['_only']):
302+
print("Adding End Time to _only")
303+
kwargs['_only'].append('End Time')
304+
300305

301306
if kwargs['_only'] == '*':
302307
kwargs.pop('_only')

0 commit comments

Comments
 (0)