From 9d73d9aae8dea68d698fb1ee707943180e62baa7 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Thu, 30 Apr 2020 11:21:57 +0200 Subject: [PATCH] added headers to tests and fixed readme. For issue #11 --- README.md | 6 ++++-- test/feature/test_lag.py | 14 ++++++++++++++ test/modules/test_feature.py | 14 ++++++++++++++ 3 files changed, 32 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 76199cb..1a50801 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,6 @@ # PyTorchTS -PyTorchTS is a [PyTorch](https://github.com/pytorch/pytorch) Probabilistic Time Series forecasting framework which provides state of the art PyTorch time series models and utilities [GluonTS](https://github.com/awslabs/gluon-ts) for loading, transforming and back-testing time series data sets. +PyTorchTS is a [PyTorch](https://github.com/pytorch/pytorch) Probabilistic Time Series forecasting framework which provides state of the art PyTorch time series models by utilizing [GluonTS](https://github.com/awslabs/gluon-ts) as its API (with minimal changes) and for loading, transforming and back-testing time series data sets. Currently the GluonTS code is copied into this repository with changes for PyTorch but eventually GluonTS should become an external requirement. ## Installation @@ -8,7 +8,9 @@ PyTorchTS is a [PyTorch](https://github.com/pytorch/pytorch) Probabilistic Time $ pip3 install pytorchts ``` -## Quick start +## Quick start (from Gluon-TS README) + +Here we highlight the ```python import matplotlib.pyplot as plt diff --git a/test/feature/test_lag.py b/test/feature/test_lag.py index bedf8d2..a56df2a 100644 --- a/test/feature/test_lag.py +++ b/test/feature/test_lag.py @@ -1,3 +1,17 @@ +# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"). +# You may not use this file except in compliance with the License. +# A copy of the License is located at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# or in the "license" file accompanying this file. This file is distributed +# on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either +# express or implied. See the License for the specific language governing +# permissions and limitations under the License. + + from pts.feature import get_lags_for_frequency # These are the expected lags for common frequencies and corner cases. diff --git a/test/modules/test_feature.py b/test/modules/test_feature.py index 47ca954..e207567 100644 --- a/test/modules/test_feature.py +++ b/test/modules/test_feature.py @@ -1,3 +1,17 @@ +# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"). +# You may not use this file except in compliance with the License. +# A copy of the License is located at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# or in the "license" file accompanying this file. This file is distributed +# on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either +# express or implied. See the License for the specific language governing +# permissions and limitations under the License. + + from itertools import chain, combinations import pytest