Add generic type for WithData trait

spatie · Nov 1, 2023 · b34caae · b34caae
1 parent 210e855
commit b34caae
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/WithData.php b/src/WithData.php
@@ -3,12 +3,17 @@
 namespace Spatie\LaravelData;
 
 use Spatie\LaravelData\Contracts\BaseData;
-use Spatie\LaravelData\Contracts\DataObject;
 use Spatie\LaravelData\Exceptions\InvalidDataClass;
 
+/**
+ * @template T
+ */
 trait WithData
 {
-    public function getData(): DataObject
+    /**
+     * @return T
+     */
+    public function getData()
     {
         $dataClass = match (true) {
             /** @psalm-suppress UndefinedThisPropertyFetch */