zhicao/dreamzero main
PYTHON
Verdict: BENIGN
Local reference
Suspicious dependency
Inferred
| 20:2 |
… allow = build_dense_allow_matrix(layout) |
|
| 21 |
assert allow.dtype == torch.bool |
Checks condition using Python assert |
| 22 |
assert allow.shape == (layout.sequence_length, layout.sequence_length) |
|
| 23 |
|
|
| 24 |
block0 = layout.slices(0) |
|
| 25 |
block1 = layout.slices(1) |
|
| 26 |
|
|
| 27 |
# Same-block observation/action are bidirectional and both can read track. |
|
| 28 |
assert allow[block0[TokenType.OBS], block0[TokenType.ACTION]].all() |
|
| 29 |
assert allow[block0[TokenType.ACTION], block0[TokenType.OBS]].all() |
|
| 30 |
assert allow[block0[TokenType.OBS], block0[TokenType.TRACK_PAST]].all() |
|
| 31 |
assert allow[block0[TokenType.ACTION], block0[TokenType.TRACK_FUTURE]].all() |
|
| 32 |
|
|
| 33 |
# Track queries never read observation, action, state, or conditioning obs. |
|
| 34 |
track_rows… |
|
| 40:2 |
… TokenType.STATE, |
|
| 41 |
) |
|
| 42 |
) |
|
| 43 |
assert not allow[track_rows][:, forbidden_columns].any() |
Checks condition using Python assert |
| 44 |
|
|
| 45 |
# Clean-past track does not consume its own noisy future target. |
|
| 46 |
assert not allow[ |
|
| 47 |
block0[TokenType.TRACK_PAST], block0[TokenType.TRACK_FUTURE] |
|
| 48 |
].any() |
|
| 49 |
assert allow[ |
|
| 50 |
block0[TokenType.TRACK_FUTURE], block0[TokenType.TRACK_PAST] |
|
| 51 |
].al… |
|
Micro-behaviors
| 20:2 |
… allow = build_dense_allow_matrix(layout) |
|
| 21 |
assert allow.dtype == torch.bool |
|
| 22 |
assert allow.shape == (layout.sequence_length, layout.sequence_length) |
|
| 23 |
|
|
| 24 |
block0 = layout.slices(0) |
|
| 25 |
block1 = layout.slices(1) |
|
| 26 |
|
|
| 27 |
# Same-block observation/action are bidirectional and both can read track. |
|
| 28 |
assert allow[block0[TokenType.OBS], block0[TokenType.ACTION]].all() |
|
| 29 |
assert allow[block0[TokenType.ACTION], block0[TokenType.OBS]].all() |
|
| 30 |
assert allow[block0[TokenType.OBS], block0[TokenType.TRACK_PAST]].all() |
|
| 31 |
assert allow[block0[TokenType.ACTION], block0[TokenType.TRACK_FUTURE]].all() |
|
| 32 |
|
|
| 33 |
# Track queries never read observation, action, state, or conditioning obs. |
|
| 34 |
track_rows… |
|
| 40:2 |
… TokenType.STATE, |
|
| 41 |
) |
|
| 42 |
) |
|
| 43 |
assert not allow[track_rows][:, forbidden_columns].any() |
|
| 44 |
|
|
| 45 |
# Clean-past track does not consume its own noisy future target. |
|
| 46 |
assert not allow[ |
|
| 47 |
block0[TokenType.TRACK_PAST], block0[TokenType.TRACK_FUTURE] |
|
| 48 |
].any() |
|
| 49 |
assert allow[ |
|
| 50 |
block0[TokenType.TRACK_FUTURE], block0[TokenType.TRACK_PAST] |
|
| 51 |
].al… |
|
Metadata
-
0
import torch
from groot.vla.model.trex_track_force.attention import (
BlockTokenLayout,
CausalWanSelfAttention,
TokenType,
build_dense_allow_matrix,
)
def test_dense_allow_matrix_exact_information_flow():
layout = BlockTokenLayout(
num_blocks=2,
conditioning_obs_tokens=2,
-
0
import torch
from groot.vla.model.trex_track_force.attention import (
BlockTokenLayout,
CausalWanSelfAttention,
TokenType,
build_dense_allow_matrix,
)
def test_dense_allow_matrix_exact_information_flow():
layout = BlockTokenLayout(
num_blocks=2,
conditioning_obs_tokens=2,
-
0
import torch
from groot.vla.model.trex_track_force.attention import (
BlockTokenLayout,
CausalWanSelfAttention,
TokenType,
build_dense_allow_matrix,
)
def test_dense_allow_matrix_exact_information_flow():
layout = BlockTokenLayout(
num_blocks=2,
conditioning_obs_tokens=2,
-
0
import torch
from groot.vla.model.trex_track_force.attention import (
BlockTokenLayout,
CausalWanSelfAttention,
TokenType,
build_dense_allow_matrix,
)
def test_dense_allow_matrix_exact_information_flow():
layout = BlockTokenLayout(
num_blocks=2,
conditioning_obs_tokens=2,
×2
Identity
| SHA-256 |
17dc10d2e5cdfc2a493a05ed8529858854019d4853df6049624d66bbbcc9545d |
| Filename |
test_attention.py |
| Package |
zhicao/dreamzero |
| Version |
main |
| PURL |
pkg:huggingface/zhicao/dreamzero@main |
Timeline
| First seen |
13 Aug 2026 21:18 UTC |
| First analyzed |
19 Aug 2026 01:56 UTC |
| Last analyzed |
19 Aug 2026 01:56 UTC |
| Last updated |
19 Aug 2026 01:56 UTC |
Labeling
| Label |
unknown |
| Label source |
forager |