From 456d8b2c8a096b244aa92760dd25d4898f10f23b Mon Sep 17 00:00:00 2001
From: Andrew DalPino <me@andrewdalpino.com>
Date: Tue, 19 Sep 2023 16:43:25 -0500
Subject: [PATCH] Get ready for release

---
 CHANGELOG.md         |  3 +++
 composer.json        | 11 +++++++----
 phpunit.xml          | 37 ++++++++++++++-----------------------
 src/Client.php       |  2 +-
 src/Helpers/JSON.php |  4 ++--
 src/RESTClient.php   |  2 +-
 6 files changed, 28 insertions(+), 31 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 783b5b8..7c7c221 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,2 +1,5 @@
+2.0.0
+    - Stable release
+
 0.0.1-beta
     - Migrate code from Server package
diff --git a/composer.json b/composer.json
index 569b50e..37dea9f 100644
--- a/composer.json
+++ b/composer.json
@@ -34,10 +34,10 @@
     },
     "require-dev": {
         "friendsofphp/php-cs-fixer": "2.18.*",
-        "phpstan/phpstan": "0.12.*",
+        "phpstan/phpstan": "^1.0",
         "phpstan/extension-installer": "^1.0",
-        "phpstan/phpstan-phpunit": "0.12.*",
-        "phpunit/phpunit": "8.5.*"
+        "phpstan/phpstan-phpunit": "^1.0",
+        "phpunit/phpunit": "^9.0"
     },
     "autoload": {
         "psr-4": {
@@ -66,7 +66,10 @@
     },
     "config": {
         "preferred-install": "dist",
-        "sort-packages": true
+        "sort-packages": true,
+        "allow-plugins": {
+            "phpstan/extension-installer": true
+        }
     },
     "funding": [
         {
diff --git a/phpunit.xml b/phpunit.xml
index 3731af3..34feae1 100644
--- a/phpunit.xml
+++ b/phpunit.xml
@@ -1,25 +1,16 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<phpunit backupGlobals="false"
-         backupStaticAttributes="false"
-         bootstrap="vendor/autoload.php"
-         colors="true"
-         convertErrorsToExceptions="true"
-         convertNoticesToExceptions="true"
-         convertWarningsToExceptions="true"
-         forceCoversAnnotation="true"
-         processIsolation="false"
-         stopOnFailure="false">
-    <testsuites>
-        <testsuite name="Base">
-            <directory>tests</directory>
-        </testsuite>
-    </testsuites>
-    <filter>
-        <whitelist processUncoveredFilesFromWhitelist="true">
-            <directory suffix=".php">src</directory>
-        </whitelist>
-    </filter>
-    <php>
-        <env name="ENV" value="testing"/>
-    </php>
+<phpunit xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" backupGlobals="false" backupStaticAttributes="false" bootstrap="vendor/autoload.php" colors="true" convertErrorsToExceptions="true" convertNoticesToExceptions="true" convertWarningsToExceptions="true" forceCoversAnnotation="true" processIsolation="false" stopOnFailure="false" xsi:noNamespaceSchemaLocation="https://schema.phpunit.de/9.3/phpunit.xsd">
+  <coverage processUncoveredFiles="true">
+    <include>
+      <directory suffix=".php">src</directory>
+    </include>
+  </coverage>
+  <testsuites>
+    <testsuite name="Base">
+      <directory>tests</directory>
+    </testsuite>
+  </testsuites>
+  <php>
+    <env name="ENV" value="testing"/>
+  </php>
 </phpunit>
diff --git a/src/Client.php b/src/Client.php
index b0605ce..8445987 100644
--- a/src/Client.php
+++ b/src/Client.php
@@ -18,7 +18,7 @@ public function predict(Dataset $dataset) : array;
      * Return the joint probabilities of each sample in a dataset.
      *
      * @param \Rubix\ML\Datasets\Dataset $dataset
-     * @return array[]
+     * @return array<array<float>>
      */
     public function proba(Dataset $dataset) : array;
 
diff --git a/src/Helpers/JSON.php b/src/Helpers/JSON.php
index b771fab..78f7c3a 100644
--- a/src/Helpers/JSON.php
+++ b/src/Helpers/JSON.php
@@ -34,7 +34,7 @@ class JSON
      *
      * @param mixed $value
      * @param int $options
-     * @param int $depth
+     * @param positive-int $depth
      * @throws \Rubix\Client\Exceptions\JSONException
      * @return string
      */
@@ -56,7 +56,7 @@ public static function encode($value, int $options = self::DEFAULT_OPTIONS, int
      *
      * @param string $data
      * @param int $options
-     * @param int $depth
+     * @param positive-int $depth
      * @throws \Rubix\Client\Exceptions\JSONException
      * @return mixed[]
      */
diff --git a/src/RESTClient.php b/src/RESTClient.php
index 409aeb2..125f35d 100644
--- a/src/RESTClient.php
+++ b/src/RESTClient.php
@@ -132,7 +132,7 @@ public function predictAsync(Dataset $dataset) : PromiseInterface
      * Return the joint probabilities of each sample in a dataset.
      *
      * @param \Rubix\ML\Datasets\Dataset $dataset
-     * @return array[]
+     * @return array<array<float>>
      */
     public function proba(Dataset $dataset) : array
     {