python_basics_setup.py

# Daniel McCray 9/12/26

"""
Python Basics — Environment Setup & Package Check (Starter File)
Complete assert_required_packages() so it asserts all required packages
are installed and importable on your machine.
"""
from __future__ import annotations
from importlib import util

REQUIRED_IMPORTS: dict[str, str] = {
"numpy": "numpy",
"pandas": "pandas",
"matplotlib": "matplotlib",
"scikit-learn": "sklearn",
"torch": "torch",
"torchvision": "torchvision",
}

def assert_required_packages(required_imports: dict[str, str] = REQUIRED_IMPORTS)-> None:
    """
    Asserts all required packages can be imported.
    - `required_imports` maps a friendly package name -> import name.
    Example: "scikit-learn" -> "sklearn"
    """
    list_of_failed_packages = []
    # 1) Try importing each module in required_imports.values()
    for value in required_imports.values():
        
        # 2) Collect any that fail
        if util.find_spec(value) is None:
            list_of_failed_packages.append(value)

    # 3) assert that none are missing (print a helpful message)
    assert list_of_failed_packages.__len__() == 0,  (
        f"Uh oh, you are missing {len(list_of_failed_packages)} required package(s): "
        f"{list_of_failed_packages}. Please refer to your Canvas assignment."
    )
   

def main() -> None:
    assert_required_packages()
    print("All required packages are installed and importable.")


if __name__ == "__main__":
    main() # Annotations