Merge pull request #6 from gurucomkz/fix-some-1

Misc updates
gurucomkz · Mar 16, 2023 · 02e195a · 02e195a
2 parents 448a35f + e768333
commit 02e195a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/EagerLoadedDataList.php b/src/EagerLoadedDataList.php
@@ -315,7 +315,7 @@ public function fulfillEagerRelations(DataObject $item)
      */
     public function getGenerator()
     {
-        $query = $this->query()->execute();
+        $query = $this->dataQuery()->execute();
 
         while ($row = $query->record()) {
             yield $this->createDataObject($row);

diff --git a/src/GridFieldEagerLoadManipulator.php b/src/GridFieldEagerLoadManipulator.php
@@ -29,7 +29,7 @@ class GridFieldEagerLoadManipulator implements GridField_DataManipulator
      */
     public function getManipulatedData(GridField $gridField, SS_List $dataList)
     {
-        /** @var DataList $dataList */
+        /** @var DataList|DataFilterEagerLoadingExtension $dataList */
         $class = $dataList->dataClass();
         $config = Config::forClass($class);
         $vars = $gridField->getForm()->getController()->getRequest()->requestVars();