Commit 7ec00c4f authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

ban numpy multithreading

parent c4ae6c7c
Loading
Loading
Loading
Loading
+2 −0
Original line number Diff line number Diff line
@@ -59,6 +59,8 @@ class Pipeline:
        if filter_warnings:
            self.filter_warnings()

        self.ban_multithreading()

    def ban_multithreading(self):
        os.environ["OMP_NUM_THREADS"] = "1"
        os.environ["MKL_NUM_THREADS"] = "1"