def create_session()

in optimum/furiosa/modeling_base.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

    def create_session(self):
        """
        Create a Furiosa runtime session for the model.

        Creates a session object using the Furiosa runtime for executing the model.

        Returns:
            None
        """
        self.sess = session.create(self.model)
        self.input_num = self.sess.input_num
        self.inputs_to_dtype = []
        for i in range(self.input_num):
            self.inputs_to_dtype.append(self.sess.input(i).dtype)