[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-09-11 12:18:35 +00:00
parent 4c2add41d7
commit 9784123463
3 changed files with 5 additions and 7 deletions
+2 -3
View File
@@ -4,10 +4,9 @@ import logging
import math import math
import os import os
from collections import defaultdict from collections import defaultdict
from collections.abc import Callable
from itertools import chain
from typing import Any, Dict, List
from collections.abc import Callable, Iterable, Mapping, Sequence from collections.abc import Callable, Iterable, Mapping, Sequence
from itertools import chain
from typing import Any
import gymnasium as gym import gymnasium as gym
import numpy as np import numpy as np
@@ -938,4 +938,4 @@ class VLAFlowMatching(nn.Module):
suffix_out = suffix_out[:, -self.config.chunk_size :] suffix_out = suffix_out[:, -self.config.chunk_size :]
suffix_out = suffix_out.to(dtype=torch.float32) suffix_out = suffix_out.to(dtype=torch.float32)
v_t = self.action_out_proj(suffix_out) v_t = self.action_out_proj(suffix_out)
return v_t return v_t
+2 -3
View File
@@ -52,14 +52,13 @@ import logging
import threading import threading
import time import time
from collections import defaultdict from collections import defaultdict
from collections.abc import Callable from collections.abc import Callable, Iterator
from contextlib import nullcontext from contextlib import nullcontext
from copy import deepcopy from copy import deepcopy
from dataclasses import asdict from dataclasses import asdict
from pathlib import Path from pathlib import Path
from pprint import pformat from pprint import pformat
from typing import Dict, List, Tuple, TypedDict from typing import TypedDict
from collections.abc import Iterator
import einops import einops
import gymnasium as gym import gymnasium as gym