Esempio n. 1
0
    def __init__(self, name, params={}, **kwargs):

        self.type = 'psutils'

        # If the user doesn't provide a file name, name based on task
        if "_save" in kwargs:
            if "file_name" not in params:
                params['file_name'] = "%s_%s.json" % (get_host().lower(),
                                                      get_user().lower())

        # Handles setting the name, setting params, and validate
        super(Task, self).__init__(name, params, **kwargs)
Esempio n. 2
0
    def __init__(self, name, params=None, **kwargs):

        if params is None:
            params = {}

        self.type = "gpu"
        self.assert_gpu()

        # If the user doesn't provide a file name, name based on task
        if "_save" in kwargs:
            if "file_name" not in params:
                params["file_name"] = "%s_%s.json" % (
                    get_host().lower(),
                    get_user().lower(),
                )

        # Handles setting the name, setting params, and validate
        super(Task, self).__init__(name, params, **kwargs)