aws_ec2_action.yaml

name: Run python test on an EC2 GPU instance

on: [push]

jobs:
  build:
    runs-on: ubuntu-latest
    steps:
    - name: Configure AWS credentials
      uses: aws-actions/configure-aws-credentials@v1
      with:
        aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }}
        aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
        aws-region: us-east-2

    - name: Create EC2 instance
      uses: kazuya-iwami/create-aws-ec2-instance@v1
      with:
        instance_type: p3.16xlarge
        ami: ami-07d0cf3af28718ef8        
      
    - name: Checkout
      uses: actions/checkout@v1
        
    - name: Set up Python 3.8
      uses: actions/setup-python@v1
      with:
        python-version: 3.8

    - name: Install dependencies
      run: |
        python -m pip install --upgrade pip
        pip install -r requirements.txt

    - name: Run test
      run: |
        pip install pytest
        pytest