fix: various fixes

Signed-off-by: tokoko <togurgenidze@gmail.com>

Author: tokoko

.github/workflows/registry-rest-api-tests.yml

@@ -145,7 +145,7 @@ jobs:
       - name: Setup and Run Registry Rest API tests
         run: |
           echo "Running Registry REST API tests..."
-          uv run pytest sdk/python/tests/integration/registration/rest_api/test_registry_rest_api.py --integration -s
+          cd sdk/python && uv run pytest tests/integration/rest_api/test_registry_rest_api.py --integration -s
 
       - name: Clean up docker images
         if: always()

Makefile

@@ -201,7 +201,7 @@ test-python-integration-local: ## Run Python integration tests (local dev mode)
 	FEAST_LOCAL_ONLINE_CONTAINER=True \
 	uv run python -m pytest --tb=short -v -n auto --color=yes --integration --durations=10 --timeout=1200 --timeout_method=thread --dist loadgroup \
 		-k "not test_lambda_materialization and not test_snowflake_materialization" \
-		-m "not rbac_remote_integration_test" \
+		-m "not rbac_remote_integration_test and not ray_offline_stores_only" \
 		--ignore=sdk/python/tests/integration/compute_engines/ray_compute \
 		--ignore=sdk/python/tests/integration/registration \
 		--log-cli-level=INFO -s \

sdk/python/tests/benchmarks/test_key_encoding_benchmarks.py

@@ -387,9 +387,9 @@ def test_performance_regression_single_entity():
         serialize_entity_key(entity_key, 3)
     elapsed = time.perf_counter() - start_time
 
-    # Should be able to do 1000 single entity serializations in < 50ms
+    # Should be able to do 1000 single entity serializations in < 100ms
     # Using a generous threshold to avoid flaky failures on CI runners
-    assert elapsed < 0.05, (
+    assert elapsed < 0.1, (
         f"Single entity serialization too slow: {elapsed:.4f}s for 1000 operations"
     )
 

sdk/python/tests/integration/online_store/test_universal_online.py

@@ -258,6 +258,9 @@ def test_write_to_online_store(environment, universal_data_sources):
         "conv_rate": [0.85],
         "acc_rate": [0.91],
         "avg_daily_trips": [14],
+        "driver_metadata": [None],
+        "driver_config": [None],
+        "driver_profile": [None],
         "event_timestamp": [pd.Timestamp(_utc_now()).round("ms")],
         "created": [pd.Timestamp(_utc_now()).round("ms")],
     }
@@ -435,6 +438,9 @@ def setup_feature_store_universal_feature_views(
         "conv_rate": [0.5, 0.3],
         "acc_rate": [0.6, 0.4],
         "avg_daily_trips": [4, 5],
+        "driver_metadata": [None, None],
+        "driver_config": [None, None],
+        "driver_profile": [None, None],
         "event_timestamp": [
             pd.to_datetime(1646263500, utc=True, unit="s"),
             pd.to_datetime(1646263600, utc=True, unit="s"),

sdk/python/tests/unit/infra/test_key_encoding_utils.py

@@ -258,8 +258,8 @@ def test_performance_bounds_single_entity():
     deserialize_time = time.perf_counter() - start
 
     # Performance bounds with generous thresholds to avoid flaky failures on CI runners
-    assert serialize_time < 0.1, f"Serialization too slow: {serialize_time:.4f}s"
-    assert deserialize_time < 0.1, f"Deserialization too slow: {deserialize_time:.4f}s"
+    assert serialize_time < 0.2, f"Serialization too slow: {serialize_time:.4f}s"
+    assert deserialize_time < 0.2, f"Deserialization too slow: {deserialize_time:.4f}s"
 
 
 def test_non_ascii_prefix_compatibility():

sdk/python/tests/utils/e2e_test_validation.py

@@ -16,16 +16,10 @@
 from tests.integration.feature_repos.universal.data_source_creator import (
     DataSourceCreator,
 )
-from tests.integration.feature_repos.universal.data_sources.bigquery import (
-    BigQueryDataSourceCreator,
-)
 from tests.integration.feature_repos.universal.data_sources.file import (
     FileDataSourceCreator,
     FileParquetDatasetSourceCreator,
 )
-from tests.integration.feature_repos.universal.data_sources.redshift import (
-    RedshiftDataSourceCreator,
-)
 
 
 def validate_offline_online_store_consistency(
@@ -225,6 +219,13 @@ def make_feature_store_yaml(
 
 # Only test if this is NOT a local test
 if os.getenv("FEAST_IS_LOCAL_TEST", "False") != "True":
+    from tests.integration.feature_repos.universal.data_sources.bigquery import (
+        BigQueryDataSourceCreator,
+    )
+    from tests.integration.feature_repos.universal.data_sources.redshift import (
+        RedshiftDataSourceCreator,
+    )
+
     NULLABLE_ONLINE_STORE_CONFIGS.extend(
         [
             IntegrationTestRepoConfig(