Update textattack/models/wrappers/remote_model_wrapper.py

Co-authored-by: Bryan Tor <[email protected]>
QData · Jan 20, 2025 · 7ccfdad · 7ccfdad
1 parent 52fe078
commit 7ccfdad
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/textattack/models/wrappers/remote_model_wrapper.py b/textattack/models/wrappers/remote_model_wrapper.py
@@ -10,11 +10,10 @@
 import transformers
 
 class RemoteModelWrapper():
-    """This model wrapper queries a remote model with a list of text inputs. 
-    
-    It sends the input to a remote endpoint provided in api_url.
-
+    """This model wrapper queries a remote model with a list of text inputs.  It sends the input to a remote endpoint provided in api_url.
 
+    Args:
+        api_url (:obj:`<TYPE HERE>`): <DESCRIPTION HERE>
     """
     def __init__(self, api_url):
         self.api_url = api_url