fix: revert changes to manual code

Author: ErinCompaan

examples/list_jobs.py

@@ -2,7 +2,7 @@
 import time
 
 start = time.time()
-tamr_client = TamrApiClient("<host-name>", [("x-api-key", "<api-key>")])
+tamr_client = TamrApiClient('<host-name>', [('x-api-key', '<api-key>')])
 one = time.time()
 
 
@@ -12,10 +12,8 @@
 response = jobs.list_jobs()
 three = time.time()
 
-response = jobs.get_job(job_id="<job-id>")
+response = jobs.get_job(job_id='<job-id>')
 four = time.time()
 
 print(response)
-print(
-    f"APIClient took {one -start}; JobsClient took {two-one} and api call took {three-two} and second is {four-three}"
-)
+print(f"APIClient took {one -start}; JobsClient took {two-one} and api call took {three-two} and second is {four-three}")

tamr_sdk/api_client.py

@@ -1,11 +1,9 @@
-from typing import List, Tuple
 from tamr_sdk.jobs.jobs_client import JobsClient
 
-
 class TamrApiClient:
-    def __init__(self, host: str, metadata: List[Tuple[str, str]]) -> None:
+    def __init__(self, host, metadata):
         self.host = host
         self.metadata = metadata
 
-    def jobs(self) -> JobsClient:
+    def jobs(self):
         return JobsClient(self.host, self.metadata)

tamr_sdk/jobs/jobs_client.py

@@ -1,36 +1,30 @@
-from typing import List, Optional, Tuple
 import grpc
 
-from google.protobuf.internal.containers import RepeatedCompositeFieldContainer
-
-import types_jobs as types
 from tamr.api.v1beta1.jobs_pb2_grpc import JobsStub
 from tamr.api.v1beta1 import jobs_pb2 as jobs
 
-
 class JobsClient:
-    def __init__(self, host: str, metadata: List[Tuple[str, str]]) -> None:
-        credentials: grpc.ChannelCredentials = grpc.ssl_channel_credentials()
-        channel: grpc.Channel = grpc.secure_channel(host, credentials)
+    def __init__(self, host, metadata):
+        credentials = grpc.ssl_channel_credentials()
+        channel = grpc.secure_channel(host, credentials)
         self.metadata = metadata
         self.stub = JobsStub(channel)
 
-    def get_job(self, job_id: str) -> types.Job:
+    def get_job(self, job_id):
         request = jobs.GetJobRequest(job_id=job_id)
         return self.stub.GetJob(request, metadata=self.metadata)
 
-    def list_jobs(
-        self, page_size: int = 10, page_token: Optional[str] = None
-    ) -> types.ListJobsResponse:
-        request = jobs.ListJobsRequest(page_size=page_size, page_token=page_token)
+    def list_jobs(self, page_size=10, page_token=None):
+        request = jobs.ListJobsRequest(
+            page_size=page_size,
+            page_token=page_token
+        )
         return self.stub.ListJobs(request, metadata=self.metadata)
 
-    def stop_job(self, job_id: str) -> types.StopJobResponse:
+    def stop_job(self, job_id):
         request = jobs.StopJobRequest(job_id=job_id)
         return self.stub.StopJob(request, metadata=self.metadata)
 
-    def create_job(
-        self, job_definition: types.CreateJobRequest
-    ) -> types.CreateJobResponse:
+    def create_job(self, job_definition):
         request = jobs.CreateJobRequest(job=job_definition)
         return self.stub.CreateJob(request, metadata=self.metadata)