Skip to content

Fix type-safety of torch.nn.Module instances #1021

Fix type-safety of torch.nn.Module instances

Fix type-safety of torch.nn.Module instances #1021

Workflow file for this run

name: Unit-tests for Conda install
on:
pull_request:
push:
branches:
- master
workflow_dispatch:
env:
CHANNEL: "nightly"
jobs:
tests:
strategy:
matrix:
python_version: ["3.8", "3.9", "3.10", "3.11"]
fail-fast: false
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux.12xlarge
repository: pytorch/captum
script: |
# Set up Environment Variables
export PYTHON_VERSION="${{ matrix.python_version }}"
# Create Conda Env
conda create -yp ci_env python="${PYTHON_VERSION}"
conda activate /pytorch/captum/ci_env
./scripts/install_via_conda.sh -n
# Run Tests
python3 -m pytest -ra --cov=. --cov-report term-missing