e9e388d4490179acb71c9f9648e65bc5862cb3f0,pynndescent/pynndescent_.py,NNDescent,__getstate__,#NNDescent#,880

Before Change

    def __getstate__(self):
        self._init_search_graph()
        self._init_search_function()
        result = self.__dict__.copy()
        result["_search_forest"] = tuple(
            [denumbaify_tree(tree) for tree in self._search_forest]
        )

After Change

        numba.set_num_threads(self._original_num_threads)

    def __getstate__(self):
        if not hasattr(self, "_search_graph"):
            self._init_search_graph()
        if not hasattr(self, "_search_function"):
            self._init_search_function()
        result = self.__dict__.copy()
        if hasattr(self, "_rp_forest"):
            del result["_rp_forest"]
        result["_search_forest"] = tuple(

Instances