fix import and clean up pre-commit

.pre-commit-config.yaml

@@ -1,21 +1,15 @@
 repos:
   - repo: local
     hooks:
-      - id: whitespace
-        name: strip whitespace
-        entry: find tinygrad -type f -name "*.py" -exec sed -i '' 's/ *$//' '{}' ';'
-        language: system
-        always_run: true
-        pass_filenames: false
       - id: ruff
         name: ruff
-        entry: ruff check .
+        entry: python3 -m ruff check .
         language: system
         always_run: true
         pass_filenames: false
       - id: mypy
         name: mypy
-        entry: mypy tinygrad/ --strict-equality
+        entry: python3 -m mypy tinygrad/ --strict-equality
         language: system
         always_run: true
         pass_filenames: false

test/models/test_real_world.py

@@ -10,7 +10,7 @@ from test.helpers import derandomize_model, is_dtype_supported
 
 from examples.gpt2 import Transformer as GPT2Transformer, MODEL_PARAMS as GPT2_MODEL_PARAMS
 from examples.hlb_cifar10 import SpeedyResNet, hyp
-from examples.llama import Transformer as LLaMaTransformer, MODEL_PARAMS as LLAMA_MODEL_PARAMS
+from examples.llama import Transformer as LLaMaTransformer
 from examples.stable_diffusion import UNetModel, unet_params
 from extra.models.unet import ResBlock