Commit 751a33dd authored by Mayank Mittal's avatar Mayank Mittal

updates pre-commit hooks

parent ca91d85f
repos: repos:
- repo: https://github.com/python/black - repo: https://github.com/python/black
rev: 22.12.0 rev: 23.1.0
hooks: hooks:
- id: black - id: black
args: ["--line-length", "120"] args: ["--line-length", "120"]
...@@ -25,7 +25,7 @@ repos: ...@@ -25,7 +25,7 @@ repos:
- id: detect-private-key - id: detect-private-key
- id: debug-statements - id: debug-statements
- repo: https://github.com/pycqa/isort - repo: https://github.com/pycqa/isort
rev: 5.11.4 rev: 5.12.0
hooks: hooks:
- id: isort - id: isort
name: isort (python) name: isort (python)
...@@ -37,12 +37,12 @@ repos: ...@@ -37,12 +37,12 @@ repos:
args: ["--py37-plus"] args: ["--py37-plus"]
# FIXME: Figure out why this is getting stuck under VPN. # FIXME: Figure out why this is getting stuck under VPN.
# - repo: https://github.com/RobertCraigie/pyright-python # - repo: https://github.com/RobertCraigie/pyright-python
# rev: v1.1.291 # rev: v1.1.296
# hooks: # hooks:
# - id: pyright # - id: pyright
# Note: We disable this by default since not all code is compatible with it. # Note: We disable this by default since not all code is compatible with it.
# - repo: https://github.com/Lucas-C/pre-commit-hooks # - repo: https://github.com/Lucas-C/pre-commit-hooks
# rev: v1.3.1 # rev: v1.4.2
# hooks: # hooks:
# - id: insert-license # - id: insert-license
# files: \.py$ # files: \.py$
......
...@@ -82,6 +82,7 @@ class FisheyeCameraCfg(PinholeCameraCfg): ...@@ -82,6 +82,7 @@ class FisheyeCameraCfg(PinholeCameraCfg):
@configclass @configclass
class UsdCameraCfg(PinholeCameraCfg.UsdCameraCfg): class UsdCameraCfg(PinholeCameraCfg.UsdCameraCfg):
"""USD related configuration of the sensor for the fisheye model."""
fisheye_nominal_width: float = None fisheye_nominal_width: float = None
"""Nominal width of fisheye lens model.""" """Nominal width of fisheye lens model."""
......
...@@ -251,7 +251,6 @@ def train(config, device): ...@@ -251,7 +251,6 @@ def train(config, device):
video_paths = None video_paths = None
rollout_check = (epoch % config.experiment.rollout.rate == 0) or (should_save_ckpt and ckpt_reason == "time") rollout_check = (epoch % config.experiment.rollout.rate == 0) or (should_save_ckpt and ckpt_reason == "time")
if config.experiment.rollout.enabled and (epoch > config.experiment.rollout.warmstart) and rollout_check: if config.experiment.rollout.enabled and (epoch > config.experiment.rollout.warmstart) and rollout_check:
# wrap model as a RolloutPolicy to prepare for rollouts # wrap model as a RolloutPolicy to prepare for rollouts
rollout_model = RolloutPolicy(model, obs_normalization_stats=obs_normalization_stats) rollout_model = RolloutPolicy(model, obs_normalization_stats=obs_normalization_stats)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment