Add types-PyYAML package and type check Python files with mypy

Author: marwan37

requirements.in

@@ -14,6 +14,7 @@ packaging
 # Tool-specific packages for ZenML extension
 watchdog
 PyYAML
+types-PyYAML
 # Conda env fixes: Ensure hash generation for typing-extensions and exceptiongroup on Python < 3.11
 typing-extensions>=4.1.0,!=4.6.3; python_version < "3.11"
 exceptiongroup==1.2.0; python_version < '3.11'

requirements.txt

@@ -81,6 +81,10 @@ pyyaml==6.0.1 \
     --hash=sha256:fd1592b3fdf65fff2ad0004b5e363300ef59ced41c2e6b3a99d4089fa8c5435d \
     --hash=sha256:fd66fc5d0da6d9815ba2cebeb4205f95818ff4b79c3ebe268e75d961704af52f
     # via -r ./requirements.in
+types-pyyaml==6.0.12.20240311 \
+    --hash=sha256:a9e0f0f88dc835739b0c1ca51ee90d04ca2a897a71af79de9aec5f38cb0a5342 \
+    --hash=sha256:b845b06a1c7e54b8e5b4c683043de0d9caf205e7434b3edc678ff2411979b8f6
+    # via -r ./requirements.in
 watchdog==4.0.0 \
     --hash=sha256:11e12fafb13372e18ca1bbf12d50f593e7280646687463dd47730fd4f4d5d257 \
     --hash=sha256:2895bf0518361a9728773083908801a376743bcc37dfa252b801af8fd281b1ca \

scripts/lint.sh

@@ -11,6 +11,10 @@ export PYTHONPATH
 echo "Linting Python files..."
 ruff bundled/tool || { echo "Linting Python files failed"; exit 1; }
 
+# Type check Python files with mypy
+echo "Type checking Python files with mypy..."
+mypy bundled/tool || { echo "Type checking Python files with mypy failed"; exit 1; }
+
 # Lint TypeScript files with eslint
 echo "Linting TypeScript files..."
 npx eslint 'src/**/*.ts' || { echo "Linting TypeScript files failed"; exit 1; }