update

Author: LongxingTan

.github/workflows/lint.yml

@@ -18,7 +18,7 @@ jobs:
           with:
             python-version: 3.8
         - name: Install Black
-          run: pip install black[jupyter]
+          run: pip install "black[jupyter]==24.8.0"
         - name: Run Black
           run: black --check .
 

.github/workflows/test.yml

@@ -21,6 +21,8 @@ jobs:
         exclude:
           - python-version: 3.9
             tf-version: 2.13.1
+          - os: macOS-latest
+            tf-version: 2.13.1
 
     steps:
       - uses: actions/checkout@v4

.pre-commit-config.yaml

@@ -19,7 +19,7 @@ repos:
     hooks:
     - id: isort
   - repo: https://github.com/psf/black
-    rev: 22.3.0
+    rev: 24.8.0
     hooks:
     - id: black
   - repo: https://github.com/nbQA-dev/nbQA

benchmark/base.py

@@ -100,9 +100,7 @@ class Dataset(ABC):
     group_column: Optional[str] = None
 
     @abstractmethod
-    def prepare_data(
-        self, **kwargs
-    ) -> Union[
+    def prepare_data(self, **kwargs) -> Union[
         Tuple[np.ndarray, np.ndarray],
         Tuple[Tuple[np.ndarray, np.ndarray], Tuple[np.ndarray, np.ndarray]],
         tf.data.Dataset,