fix: pass input_data in xgboost integration test

test/integration/api/test_main_api.py

@@ -166,8 +166,6 @@ def test_pandas_input_for_api():
 
     train_features = pd.DataFrame(train_data.features)
     train_target = pd.Series(train_data.target.reshape(-1))
-
-    test_features = pd.DataFrame(test_data.features)
     test_target = pd.Series(test_data.target.reshape(-1))
 
     # task selection, initialisation of the framework
@@ -177,7 +175,7 @@ def test_pandas_input_for_api():
     baseline_model.fit(features=train_features, target=train_target, predefined_model='xgboost')
 
     # evaluate the prediction with test data
-    prediction = baseline_model.predict(features=test_features)
+    prediction = baseline_model.predict(features=test_data)
 
     assert len(prediction) == len(test_target)