Skip to content

Commit 3031245

Browse files
author
Bhargav Dodla
committed
fix: Removed unused logging and methods
Signed-off-by: Bhargav Dodla <bdodla@expediagroup.com>
1 parent 92b62aa commit 3031245

File tree

4 files changed

+3
-59
lines changed

4 files changed

+3
-59
lines changed

sdk/python/feast/infra/online_stores/remote.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -112,9 +112,8 @@ def online_read(
112112
result_tuples.append((event_ts, feature_values_dict))
113113
return result_tuples
114114
else:
115-
error_msg = f"Unable to retrieve the online store data using feature server API. Error_code={response.status_code}, error_message={response.reason}"
115+
error_msg = f"Unable to retrieve the online store data using feature server API. Error_code={response.status_code}, error_message={response.text}"
116116
logger.error(error_msg)
117-
logger.error(f"Error response Text: {response.text}")
118117
raise RuntimeError(error_msg)
119118

120119
def _construct_online_read_api_json_request(

sdk/python/feast/infra/registry/snowflake.py

+1-25
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from datetime import datetime, timedelta, timezone
66
from enum import Enum
77
from threading import Lock
8-
from typing import Any, Callable, List, Literal, Optional, Set, Union
8+
from typing import Any, Callable, List, Literal, Optional, Union
99

1010
from pydantic import ConfigDict, Field, StrictStr
1111

@@ -1101,30 +1101,6 @@ def process_project(project: Project):
11011101

11021102
return r
11031103

1104-
def _get_all_projects(self) -> Set[str]:
1105-
projects = set()
1106-
1107-
base_tables = [
1108-
"DATA_SOURCES",
1109-
"ENTITIES",
1110-
"FEATURE_VIEWS",
1111-
"ON_DEMAND_FEATURE_VIEWS",
1112-
"STREAM_FEATURE_VIEWS",
1113-
"PERMISSIONS",
1114-
]
1115-
1116-
with GetSnowflakeConnection(self.registry_config) as conn:
1117-
for table in base_tables:
1118-
query = (
1119-
f'SELECT DISTINCT project_id FROM {self.registry_path}."{table}"'
1120-
)
1121-
df = execute_snowflake_statement(conn, query).fetch_pandas_all()
1122-
1123-
for row in df.iterrows():
1124-
projects.add(row[1]["PROJECT_ID"])
1125-
1126-
return projects
1127-
11281104
def _get_last_updated_metadata(self, project: str):
11291105
with GetSnowflakeConnection(self.registry_config) as conn:
11301106
query = f"""

sdk/python/feast/infra/registry/sql.py

+1-19
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from datetime import datetime, timezone
55
from enum import Enum
66
from pathlib import Path
7-
from typing import Any, Callable, Dict, List, Optional, Set, Union
7+
from typing import Any, Callable, Dict, List, Optional, Union
88

99
from pydantic import StrictInt, StrictStr
1010
from sqlalchemy import ( # type: ignore
@@ -1043,24 +1043,6 @@ def _get_last_updated_metadata(self, project: str):
10431043

10441044
return datetime.fromtimestamp(update_time, tz=timezone.utc)
10451045

1046-
def _get_all_projects(self) -> Set[str]:
1047-
projects = set()
1048-
with self.engine.begin() as conn:
1049-
for table in {
1050-
entities,
1051-
data_sources,
1052-
feature_views,
1053-
on_demand_feature_views,
1054-
stream_feature_views,
1055-
permissions,
1056-
}:
1057-
stmt = select(table)
1058-
rows = conn.execute(stmt).all()
1059-
for row in rows:
1060-
projects.add(row._mapping["project_id"])
1061-
1062-
return projects
1063-
10641046
def _get_permission(self, name: str, project: str) -> Permission:
10651047
return self._get_object(
10661048
table=permissions,

sdk/python/tests/unit/permissions/auth/server/test_auth_registry_server.py

-13
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import logging
21
from datetime import datetime
32

43
import assertpy
@@ -28,8 +27,6 @@
2827
from tests.utils.auth_permissions_util import get_remote_registry_store
2928
from tests.utils.http_server import check_port_open # noqa: E402
3029

31-
logger = logging.getLogger(__name__)
32-
3330

3431
@pytest.fixture
3532
def start_registry_server(
@@ -39,7 +36,6 @@ def start_registry_server(
3936
feature_store,
4037
monkeypatch,
4138
):
42-
logger.info(f"Starting Registry call at {server_port}")
4339
if "kubernetes" in auth_config:
4440
mock_utils.mock_kubernetes(request=request, monkeypatch=monkeypatch)
4541
elif "oidc" in auth_config:
@@ -53,29 +49,23 @@ def start_registry_server(
5349
assertpy.assert_that(server_port).is_not_equal_to(0)
5450

5551
print(f"Starting Registry at {server_port}")
56-
logger.info(f"{auth_config}: {server_port}: Starting Registry at {server_port}")
5752
server = start_server(
5853
feature_store,
5954
server_port,
6055
wait_for_termination=False,
6156
)
6257
print("Waiting server availability")
63-
logger.info(f"{auth_config}: {server_port}: Waiting server availability")
6458
wait_retry_backoff(
6559
lambda: (None, check_port_open("localhost", server_port)),
6660
timeout_secs=10,
6761
)
6862
print("Server started")
69-
logger.info(f"{auth_config}: {server_port}: Server started")
7063

7164
yield server
7265

7366
print("Stopping server")
74-
logger.info(f"{auth_config}: {server_port}: Stopping server")
7567
server.stop(grace=None) # Teardown server
7668

77-
# server.wait_for_termination()
78-
7969

8070
def test_registry_apis(
8171
auth_config,
@@ -86,9 +76,6 @@ def test_registry_apis(
8676
applied_permissions,
8777
):
8878
print(f"Running for\n:{auth_config}")
89-
logger.info(
90-
f"Running Perm Tests for\n:{auth_config} : {applied_permissions}: {server_port}"
91-
)
9279
remote_feature_store = get_remote_registry_store(server_port, feature_store)
9380
permissions = _test_list_permissions(remote_feature_store, applied_permissions)
9481
_test_get_entity(remote_feature_store, applied_permissions)

0 commit comments

Comments
 (0)