diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
index f2295d07a..8787d7727 100644
--- a/.github/workflows/lint.yml
+++ b/.github/workflows/lint.yml
@@ -1,14 +1,7 @@
 ---
 name: Lint
-on: [push, pull_request] # yamllint disable-line rule:truthy
+on: [push, pull_request] 
 jobs:
-  yaml-lint:
-    runs-on: ubuntu-latest
-    steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
-      - run: pip install yamllint
-      - run: yamllint .
   flake8-lint:
     runs-on: ubuntu-latest
     steps:
diff --git a/byte.py b/byte.py
index d0a5a92f2..23c2166c1 100644
--- a/byte.py
+++ b/byte.py
@@ -5,6 +5,7 @@
 import numpy as np
 import torch
 import struct
+import magic
 
 
 from enum import auto