Merge pull request #348 from runpod/black

feat: use 'black' to unify formatting

Author: ef0xa

examples/api/create_endpoint.py

@@ -8,9 +8,7 @@
 try:
 
     new_template = runpod.create_template(
-        name="test",
-        image_name="runpod/base:0.4.4",
-        is_serverless=True
+        name="test", image_name="runpod/base:0.4.4", is_serverless=True
     )
 
     print(new_template)
@@ -21,7 +19,7 @@
         gpu_ids="AMPERE_16",
         workers_min=0,
         workers_max=1,
-        flashboot=True
+        flashboot=True,
     )
 
     print(new_endpoint)

examples/api/create_template.py

@@ -7,10 +7,7 @@
 
 try:
 
-    new_template = runpod.create_template(
-        name="test",
-        image_name="runpod/base:0.1.0"
-    )
+    new_template = runpod.create_template(name="test", image_name="runpod/base:0.1.0")
 
     print(new_template)
 

examples/api/get_endpoints.py

@@ -1,6 +1,5 @@
 """ Get all endpoints from the API """
 
-
 import runpod
 
 endpoints = runpod.get_endpoints()

examples/endpoints/asyncio_job_request.py

@@ -7,15 +7,17 @@
 import runpod
 from runpod import http_client, AsyncioEndpoint, AsyncioJob
 
-asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())  # For Windows Users
+asyncio.set_event_loop_policy(
+    asyncio.WindowsSelectorEventLoopPolicy()
+)  # For Windows Users
 
 runpod.api_key = "YOUR_API_KEY"
 
 
 async def main():
-    '''
+    """
     Function to run the example.
-    '''
+    """
     async with http_client.AsyncClientSession() as session:
         # Invoke API
         payload = {}
@@ -34,4 +36,5 @@ async def main():
         # Print output
         print(output)
 
+
 asyncio.run(main())

examples/endpoints/run.py

@@ -1,6 +1,6 @@
-'''
+"""
 Example of calling an endpoint using the RunPod Python Language Library.
-'''
+"""
 
 import runpod
 
@@ -9,11 +9,9 @@
 
 endpoint = runpod.Endpoint("sdxl")  # Where "sdxl" is the endpoint ID
 
-run_request = endpoint.run({
-    "input": {
-        "prompt": "a photo of a horse the size of a Boeing 787"
-    }
-})
+run_request = endpoint.run(
+    {"input": {"prompt": "a photo of a horse the size of a Boeing 787"}}
+)
 
 # Check the status of the run request
 print(run_request.status())

examples/endpoints/run_sync.py

@@ -1,6 +1,6 @@
-'''
+"""
 Example of calling an endpoint using the RunPod Python Language Library.
-'''
+"""
 
 import runpod
 
@@ -12,12 +12,8 @@
 try:
     # Run the endpoint synchronously, blocking until the endpoint run is complete.
     run_request = endpoint.run_sync(
-        {
-            "input": {
-                "prompt": "a photo of a horse the size of a Boeing 787"
-            }
-        },
-        timeout=60  # Seconds
+        {"input": {"prompt": "a photo of a horse the size of a Boeing 787"}},
+        timeout=60,  # Seconds
     )
 
     print(run_request)

examples/endpoints/streaming.py

@@ -7,12 +7,14 @@
 
 endpoint = runpod.Endpoint("gwp4kx5yd3nur1")
 
-run_request = endpoint.run({
-    "input": {
-        "mock_return": ["a", "b", "c", "d", "e", "f", "g"],
-        "mock_delay": 1,
+run_request = endpoint.run(
+    {
+        "input": {
+            "mock_return": ["a", "b", "c", "d", "e", "f", "g"],
+            "mock_delay": 1,
+        }
     }
-})
+)
 
 for output in run_request.stream():
     print(output)

examples/graphql_wrapper.py

@@ -1,6 +1,6 @@
-''''
+"""'
 GraphQL wrapper for the RunPod API
-'''
+"""
 
 import time
 import runpod

examples/serverless/concurrent_handler.py

@@ -5,12 +5,14 @@
 
 # ---------------------------------- Handler --------------------------------- #
 async def async_generator_handler(job):
-    '''
+    """
     Async generator type handler.
-    '''
+    """
     return job
 
 
-runpod.serverless.start({
-    "handler": async_generator_handler,
-})
+runpod.serverless.start(
+    {
+        "handler": async_generator_handler,
+    }
+)

examples/serverless/logger.py

@@ -1,15 +1,15 @@
-""" Example of using the RunPodLogger class. """""
+""" Example of using the RunPodLogger class. """ ""
 
 import runpod
 
-JOB_ID = '1234567890'
+JOB_ID = "1234567890"
 log = runpod.RunPodLogger()
 
 
-log.debug('A debug message')
-log.info('An info message')
-log.warn('A warning message')
-log.error('An error message')
+log.debug("A debug message")
+log.info("An info message")
+log.warn("A warning message")
+log.error("An error message")
 
 # Output:
 # DEBUG   | A debug message
@@ -18,10 +18,10 @@
 # ERROR   | An error message
 
 
-log.debug('A debug message', request_id=JOB_ID)
-log.info('An info message', request_id=JOB_ID)
-log.warn('A warning message', request_id=JOB_ID)
-log.error('An error message', request_id=JOB_ID)
+log.debug("A debug message", request_id=JOB_ID)
+log.info("An info message", request_id=JOB_ID)
+log.warn("A warning message", request_id=JOB_ID)
+log.error("An error message", request_id=JOB_ID)
 
 # Output:
 # {"requestId": "1234567890", "message": "A debug message", "level": "DEBUG"}