@@ -168,7 +168,7 @@ def create_model(self, name, task, model_file_path, model_class, dependencies={}
168168 :param str name: Name of the model, which must be unique across all models added by the current user
169169 :param str task: Task associated with the model, where the model must adhere to the specification of the task
170170 :param str model_file_path: Path to a single Python file that contains the definition for the model class
171- :param obj model_class: The name of the model class inside the Python file. This class should implement :class:`rafiki.model.BaseModel`
171+ :param str model_class: The name of the model class inside the Python file. This class should implement :class:`rafiki.model.BaseModel`
172172 :param dependencies: List of dependencies & their versions
173173 :type dependencies: dict[str, str]
174174 :param access_right: Model access right
@@ -188,7 +188,7 @@ def create_model(self, name, task, model_file_path, model_class, dependencies={}
188188 ``<dependency_name>`` corresponds to the name of the Python Package Index (PyPI) package (e.g. ``tensorflow``)
189189 and ``<dependency_version>`` corresponds to the version of the PyPI package (e.g. ``1.12.0``). These dependencies
190190 will be lazily installed on top of the worker's Docker image before the submitted model's code is executed.
191- If the model is to be run on GPU, Rafiki would map dependencies to their GPU-supported versions, if required .
191+ If the model is to be run on GPU, Rafiki would map dependencies to their GPU-supported versions, if supported .
192192 For example, ``{ 'tensorflow': '1.12.0' }`` will be installed as ``{ 'tensorflow-gpu': '1.12.0' }``.
193193 Rafiki could also parse specific dependency names to install certain non-PyPI packages.
194194 For example, ``{ 'singa': '1.1.1' }`` will be installed as ``singa-cpu=1.1.1`` or ``singa-gpu=1.1.1`` using ``conda``.
@@ -583,7 +583,7 @@ def stop_inference_job(self, app, app_version=-1):
583583 # Advisors
584584 ####################################
585585
586- def create_advisor (self , knob_config_str , advisor_id = None ):
586+ def _create_advisor (self , knob_config_str , advisor_id = None ):
587587 '''
588588 Creates a Rafiki advisor. If `advisor_id` is passed, it will create an advisor
589589 of that ID, or do nothing if an advisor of that ID has already been created.
@@ -600,7 +600,7 @@ def create_advisor(self, knob_config_str, advisor_id=None):
600600 })
601601 return data
602602
603- def generate_proposal (self , advisor_id ):
603+ def _generate_proposal (self , advisor_id ):
604604 '''
605605 Generate a proposal of knobs from an advisor.
606606
@@ -611,7 +611,7 @@ def generate_proposal(self, advisor_id):
611611 data = self ._post ('/advisors/{}/propose' .format (advisor_id ), target = 'advisor' )
612612 return data
613613
614- def feedback_to_advisor (self , advisor_id , knobs , score ):
614+ def _feedback_to_advisor (self , advisor_id , knobs , score ):
615615 '''
616616 Feedbacks to the advisor on the score of a set of knobs.
617617 Additionally returns another proposal of knobs after ingesting feedback.
@@ -629,7 +629,7 @@ def feedback_to_advisor(self, advisor_id, knobs, score):
629629 })
630630 return data
631631
632- def delete_advisor (self , advisor_id ):
632+ def _delete_advisor (self , advisor_id ):
633633 '''
634634 Deletes a Rafiki advisor.
635635
0 commit comments