Appease overbearing Stan

Author: andrewdalpino

composer.json

@@ -38,7 +38,7 @@
         "amphp/parallel": "^1.3",
         "psr/log": "^1.1",
         "rubix/tensor": "^3.0",
-        "scienide/okbloomer": "^1.0",
+        "andrewdalpino/okbloomer": "^1.0",
         "symfony/polyfill-mbstring": "^1.0",
         "symfony/polyfill-php80": "^1.17",
         "wamania/php-stemmer": "^2.0"

src/Transformers/IntervalDiscretizer.php

@@ -47,7 +47,7 @@ class IntervalDiscretizer implements Transformer, Stateful, Persistable
     /**
      * The bin intervals of the fitted data.
      *
-     * @var array<float[]>|null
+     * @var array<(int|float)[]>|null
      */
     protected ?array $intervals = null;
 
@@ -92,7 +92,7 @@ public function fitted() : bool
     /**
      * Return the bin intervals of the fitted data.
      *
-     * @return array<float[]>|null
+     * @return array<(int|float)[]>|null
      */
     public function intervals() : ?array
     {

src/Transformers/MinMaxNormalizer.php

@@ -137,7 +137,10 @@ public function fit(Dataset $dataset) : void
             if ($type->isContinuous()) {
                 $values = $dataset->feature($column);
 
+                /** @var int|float $min */
                 $min = min($values);
+
+                /** @var int|float $max */
                 $max = max($values);
 
                 $scale = ($this->max - $this->min) / (($max - $min) ?: EPSILON);
@@ -167,7 +170,10 @@ public function update(Dataset $dataset) : void
         foreach ($this->scales as $column => &$scale) {
             $values = $dataset->feature($column);
 
+            /** @var int|float $min */
             $min = min($this->minimums[$column], ...$values);
+
+            /** @var int|float $max */
             $max = max($this->maximums[$column], ...$values);
 
             $scale = ($this->max - $this->min) / (($max - $min) ?: EPSILON);

src/Transformers/OneHotEncoder.php

@@ -37,7 +37,7 @@ class OneHotEncoder implements Transformer, Stateful, Persistable
     /**
      * The set of unique possible categories per feature column of the training set.
      *
-     * @var array<string[]>|null
+     * @var array<int[]>|null
      */
     protected ?array $categories = null;
 
@@ -90,7 +90,10 @@ public function fit(Dataset $dataset) : void
 
                 $categories = array_values(array_unique($values));
 
-                $this->categories[$column] = array_flip($categories);
+                /** @var int[] $offsets */
+                $offsets = array_flip($categories);
+
+                $this->categories[$column] = $offsets;
             }
         }
     }