fix #720

Author: dimitri-yatsenko

datajoint/autopopulate.py

@@ -5,9 +5,8 @@
 import random
 import inspect
 from tqdm import tqdm
-from pymysql import OperationalError
 from .expression import QueryExpression, AndList, U
-from .errors import DataJointError
+from .errors import DataJointError, LostConnectionError
 from .table import FreeTable
 import signal
 
@@ -161,7 +160,7 @@ def handler(signum, frame):
                     except (KeyboardInterrupt, SystemExit, Exception) as error:
                         try:
                             self.connection.cancel_transaction()
-                        except OperationalError:
+                        except LostConnectionError:
                             pass
                         error_message = '{exception}{msg}'.format(
                             exception=error.__class__.__name__,

datajoint/table.py

@@ -127,18 +127,18 @@ def parents(self, primary=None):
 
     def children(self, primary=None):
         """
-        :param primary: if None, then all parents are returned. If True, then only foreign keys composed of
+        :param primary: if None, then all children are returned. If True, then only foreign keys composed of
             primary key attributes are considered.  If False, the only foreign keys including at least one non-primary
             attribute are considered.
         :return: dict of tables with foreign keys referencing self
         """
         return self.connection.dependencies.children(self.full_table_name, primary)
 
     def descendants(self):
-        return self. connection.dependencies.descendants(self.full_table_name)
+        return self.connection.dependencies.descendants(self.full_table_name)
 
     def ancestors(self):
-        return self. connection.dependencies.ancestors(self.full_table_name)
+        return self.connection.dependencies.ancestors(self.full_table_name)
 
     @property
     def is_declared(self):