small fixes to highlighting

uiua-lang · Nov 27, 2024 · 4a6dbb1 · 4a6dbb1
1 parent 2f9dd49
commit 4a6dbb1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/src/compile/mod.rs b/src/compile/mod.rs
@@ -330,7 +330,7 @@ impl Compiler {
         let input: EcoString = fs::read_to_string(path)
             .map_err(|e| UiuaErrorKind::Load(path.into(), e.into()))?
             .into();
-        _ = crate::lsp::Spans::from_input(&input);
+        // _ = crate::lsp::Spans::from_input(&input);
         self.asm.inputs.files.insert(path.into(), input.clone());
         self.load_impl(&input, InputSrc::File(path.into()))
     }

diff --git a/src/primitive/mod.rs b/src/primitive/mod.rs
@@ -494,6 +494,7 @@ impl Primitive {
             (Couple, None) => Signature::new(2, 1),
             (Box, None) => Signature::new(1, 1),
             (Transpose | Sqrt | Round | Floor | Ceil | Rand | Utf8, _) => return self.sig(),
+            (Stack, Some(n)) if n >= 0 => Signature::new(n as usize, n as usize),
             _ => return None,
         })
     }
@@ -2567,9 +2568,6 @@ mod tests {
         {{
             "include": "#module_delim"
         }},
-		{{
-			"include": "#numbers"
-		}},
         {{
             "include": "#strand"
         }},
@@ -2593,7 +2591,10 @@ mod tests {
 		}},
         {{
             "include": "#idents"
-        }}
+        }},
+		{{
+			"include": "#numbers"
+		}}
 	],
 	"repository": {{
         "idents": {{