Method setUp

axlearn/common/test_utils.py:170–173

(self)

        return "FAKE"

    def setUp(self):
        super().setUp()
        push_data_dir(self.data_dir)
        utils_spmd.setup(jax_backend=self._jax_backend())

    def _jax_backend(self) -> str:
        # Setup without distributed initialization by default.