diff --git a/crates/challenge-agentic/src/sim.rs b/crates/challenge-agentic/src/sim.rs index 8fff93cfd..7b6e7e308 100644 --- a/crates/challenge-agentic/src/sim.rs +++ b/crates/challenge-agentic/src/sim.rs @@ -11,8 +11,8 @@ use std::path::Path; use async_trait::async_trait; use challenge_ast::{ - fingerprint_source, similarity_bps, static_source_cheat, top_k_nearest, AST_CHEAT_BPS, - AST_SUSPICIOUS_BPS, BASELINE_CORPUS_PREFIX, SourceCheatKind, + fingerprint_source, similarity_bps, static_source_cheat, top_k_nearest, SourceCheatKind, + AST_CHEAT_BPS, AST_SUSPICIOUS_BPS, BASELINE_CORPUS_PREFIX, }; use serde_json::Value; diff --git a/crates/challenge-ast/src/source_cheats.rs b/crates/challenge-ast/src/source_cheats.rs index a0f45f32b..5ed2e76b8 100644 --- a/crates/challenge-ast/src/source_cheats.rs +++ b/crates/challenge-ast/src/source_cheats.rs @@ -89,9 +89,7 @@ fn noncausal_seq_mix_reason(architecture_py: &str) -> Option<&'static str> { || src.contains("transpose(-1, -2)") || src.contains("transpose(-2, -1)") || src.contains(".mT") - || (src.contains("einops.rearrange") - && src.contains("b t d") - && src.contains("b d t")); + || (src.contains("einops.rearrange") && src.contains("b t d") && src.contains("b d t")); if !has_time_transpose { return None;