Skip to content

Commit 914e527

Browse files
committed
fix to normalize entityset
1 parent 4dd5df1 commit 914e527

File tree

2 files changed

+83
-1190
lines changed

2 files changed

+83
-1190
lines changed

autonormalize/autonormalize.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -164,10 +164,10 @@ def normalize_entity(es, accuracy=0.98):
164164
# TO DO: add option to pass an EntitySet with more than one entity, and specify which one
165165
# to normalize while preserving existing relationships
166166

167-
if len(es.entities()) > 1:
167+
if len(es.entities) > 1:
168168
raise ValueError('There is more than one entity in this EntitySet')
169-
if len(es.entities()) == 0:
169+
if len(es.entities) == 0:
170170
raise ValueError('This EntitySet is empty')
171-
entity = es.entities()[0]
171+
entity = es.entities[0]
172172
new_es = auto_entityset(entity.df, accuracy, index=entity.index, name=es.id, time_index=entity.time_index)
173173
return new_es

0 commit comments

Comments
 (0)