Skip to content

Change torch state_dict file name to prediction-net-state.pt #945

Change torch state_dict file name to prediction-net-state.pt

Change torch state_dict file name to prediction-net-state.pt #945

Workflow file for this run

name: Doctests
on: [push, pull_request]
jobs:
test:
strategy:
max-parallel: 4
fail-fast: false
matrix:
python-version: ['3.9']
platform: [ubuntu-latest]
runs-on: ${{ matrix.platform }}
steps:
- uses: actions/checkout@v3
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v4
with:
python-version: ${{ matrix.python-version }}
- name: Install other dependencies
run: |
python -m pip install -U pip
pip install -e .
pip install -r requirements/requirements-test.txt
pip install -r requirements/requirements-arrow.txt
- name: Test with pytest
run: |
pytest -n2 --doctest-modules \
src/gluonts/core \
src/gluonts/dataset \
src/gluonts/model \
src/gluonts/time_feature \
src/gluonts/zebras \
src/gluonts/maybe.py \
src/gluonts/itertools.py