Fixes to array translation

getzlab · Jun 27, 2019 · ff99518 · ff99518
1 parent 33c0d38
commit ff99518
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/dalmatian/base.py b/dalmatian/base.py
@@ -1589,7 +1589,7 @@ def update_entity_attributes(self, etype, attrs):
                                 'entityType': reserved_attrs[i],
                                 'entityName': str(j)
                             } if i in reserved_attrs else str(j))
-                        } for i,j in row.iteritems() if not pd.isnull(j)
+                        } for i,j in row.iteritems() if not np.any(pd.isnull(j))
                     ]
                 }])
         elif isinstance(attrs, pd.Series):
@@ -1606,7 +1606,7 @@ def update_entity_attributes(self, etype, attrs):
                         } if attrs.name in reserved_attrs else str(j))
                     }
                 ]
-            } for i,j in attrs.iteritems() if not pd.isnull(j)]
+            } for i,j in attrs.iteritems() if not np.any(pd.isnull(j))]
         else:
             raise ValueError('Unsupported input format.')
 

diff --git a/dalmatian/wmanager.py b/dalmatian/wmanager.py
@@ -494,7 +494,7 @@ def __df_upload_translation_layer_internal(self, func, etype, updates, *args):
                         break
                 if name not in selector:
                     # Apply the manual translated update to this row
-                    translations |= self.__df_upload_translation_layer_internal(func, workspace, etype, data)
+                    translations |= self.__df_upload_translation_layer_internal(func, etype, data)
             func(etype, updates.loc[[*selector]], *args)
         else:
             response = self.__patch(