fix a bug in repeat signature inference

changelog.md

@@ -3,6 +3,10 @@
 Uiua is not yet stable.
 <!-- This version is not yet released. If you are reading this on the website, then these changes are live here. -->
 
+## 0.10.2 - 2024-04-08
+### Interpreter
+- Fix a bug in [`repeat ⍥`](https://uiua.org/docs/repeat) with [`infinity ∞`](https://uiua.org/docs/infinity) signature inference
+
 ## 0.10.1 - 2024-04-07
 ### Interpreter
 - [`under ⍜`](https://uiua.org/docs/under) of pattern matching now works correctly

src/check.rs

@@ -361,21 +361,20 @@ impl<'a> VirtualEnv<'a> {
  }
  } else if n.is_infinite() {
  match sig.args.cmp(&sig.outputs) {
- Ordering::Equal => {}
  Ordering::Greater => {
  return Err(SigCheckError::from(format!(
  "repeat with infinity and a function with signature {sig}"
  ))
  .loop_overreach());
  }
- Ordering::Less => {
+ Ordering::Less if self.array_stack.is_empty() => {
  return Err(SigCheckError::from(format!(
  "repeat with infinity and a function with signature {sig}"
  ))
  .loop_excess());
  }
+ _ => self.handle_sig(sig)?,
  }
- self.handle_sig(sig)?;
  } else {
  return Err("repeat without a natural number or infinity".into());
  }

tests/signature.ua

@@ -28,6 +28,8 @@ F ← |3 ⍜(↙⊙↘)⇌
 F ← |3 ⍜∩°□(⊂⊂)
 # Try
 F ← |1 ⍣⋕∘
+# Repeat
+F ← |1 {⍥(⊜⧻..)∞}
 # Do
 C ← ⟨+1×3|÷2⟩=0◿2.
 F ← |1 ◌⍢(|2 ⊂)(|1.3 ¬∊,,C⊢.)