diff --git a/spanner_graphs/database.py b/spanner_graphs/database.py index 176cf20..a4cf505 100644 --- a/spanner_graphs/database.py +++ b/spanner_graphs/database.py @@ -86,7 +86,7 @@ def mock(cls) -> 'DatabaseSelector': return cls(env=SpannerEnv.MOCK) @classmethod - def experimental_host( + def for_experimental_host( cls, experimental_host: str, database: str, use_plain_text: bool = False, ca_certificate: str | None = None, client_certificate: str | None = None, client_key: str | None = None ) -> "DatabaseSelector": """Creates a selector for a Google Experimental Host Spanner database.""" diff --git a/spanner_graphs/graph_server.py b/spanner_graphs/graph_server.py index 7d9732d..fb79583 100644 --- a/spanner_graphs/graph_server.py +++ b/spanner_graphs/graph_server.py @@ -66,7 +66,7 @@ def dict_to_selector(selector_dict: Dict[str, Any]) -> DatabaseSelector: elif env == SpannerEnv.MOCK: return DatabaseSelector.mock() elif env == SpannerEnv.EXPERIMENTAL_HOST: - return DatabaseSelector.experimental_host( + return DatabaseSelector.for_experimental_host( selector_dict["experimental_host"], selector_dict["database"], selector_dict["use_plain_text"], selector_dict["ca_certificate"], selector_dict["client_certificate"], selector_dict["client_key"] ) raise ValueError(f"Invalid env in selector dict: {selector_dict}") diff --git a/spanner_graphs/magics.py b/spanner_graphs/magics.py index db2155a..4754ab7 100644 --- a/spanner_graphs/magics.py +++ b/spanner_graphs/magics.py @@ -247,7 +247,7 @@ def spanner_graph(self, line: str, cell: str): if bool(args.client_certificate) != bool(args.client_key): raise ValueError("client_certificate and client_key must both be provided together.") - selector = DatabaseSelector.experimental_host( + selector = DatabaseSelector.for_experimental_host( experimental_host=args.experimental_host, database=args.database, use_plain_text=args.use_plain_text, ca_certificate=args.ca_certificate, client_certificate=args.client_certificate, client_key=args.client_key ) else: