diff --git a/CHANGELOG.rst b/CHANGELOG.rst index 8500991d..03109549 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -3,7 +3,6 @@ Unreleased * feat: Lowercase the ``null_values`` provided to individual data types, since all comparisons to ``null_values`` are case-insensitive. (#770) * feat: :class:`.Mean` works with :class:`.TimeDelta`. (#761) -* fix: Allow consecutive calls to :meth:`.Table.group_by`. (#765) 1.7.1 - Jan 4, 2023 ------------------- diff --git a/agate/tableset/__init__.py b/agate/tableset/__init__.py index 7e70236f..4876d00b 100644 --- a/agate/tableset/__init__.py +++ b/agate/tableset/__init__.py @@ -148,12 +148,7 @@ def _proxy(self, method_name, *args, **kwargs): tables = [] for key, table in self.items(): - result = getattr(table, method_name)(*args, **kwargs) - if isinstance(result, TableSet): - for table in result.values(): - tables.append(table) - else: - tables.append(result) + tables.append(getattr(table, method_name)(*args, **kwargs)) return self._fork( tables, diff --git a/tests/test_table/test_group_by.py b/tests/test_table/test_group_by.py index 6178a3a8..cb51ffd4 100644 --- a/tests/test_table/test_group_by.py +++ b/tests/test_table/test_group_by.py @@ -108,8 +108,3 @@ def test_group_by_bad_column(self): with self.assertRaises(KeyError): table.group_by('bad') - - def test_group_by_twice(self): - table = Table(self.rows, self.column_names, self.column_types) - - repr(table.group_by('one').group_by('two'))