Method tree_flatten

axlearn/common/checkpointer_test.py:1189–1191

(self)

            def tree_flatten_with_keys(self):
                raise NotImplementedError

            def tree_flatten(self):
                # Convert dict_values and_keys to lists to avoid holding reference to the VDict.
                return (list(self.values()), list(self.keys()))