Skip to content
This repository was archived by the owner on Oct 31, 2023. It is now read-only.

Commit b312b08

Browse files
authored
Merge pull request #41 from EntilZha/pr-passthrough-rescaler
Add passthrough rescaler + Git CI style checker
2 parents 1160422 + faca41e commit b312b08

File tree

3 files changed

+30
-0
lines changed

3 files changed

+30
-0
lines changed

.circleci/ci-requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
yapf

.circleci/config.yml

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
version: 2.1
2+
orbs:
3+
python: circleci/[email protected]
4+
5+
jobs:
6+
linting:
7+
executor: python/default
8+
steps:
9+
- checkout
10+
- python/install-packages:
11+
pip-dependency-file: .circleci/ci-requirements.txt
12+
pkg-manager: pip
13+
- run:
14+
name: Run tests
15+
command: yapf -d --recursive rlmeta
16+
17+
workflows:
18+
main:
19+
jobs:
20+
- linting

rlmeta/core/rescalers.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,15 @@ def recover(self, x: torch.Tensor) -> torch.Tensor:
3131
"""
3232

3333

34+
class PassthroughRescaler(Rescaler):
35+
36+
def rescale(self, x: torch.Tensor) -> torch.Tensor:
37+
return x
38+
39+
def recover(self, x: torch.Tensor) -> torch.Tensor:
40+
return x
41+
42+
3443
class RMSRescaler(Rescaler):
3544

3645
def __init__(self, size: Union[int, Tuple[int]]) -> None:

0 commit comments

Comments
 (0)