TypeError in function_optimization/Function_2D.ipynb

[Br0kenSmi1e]

I was trying to run the examples given in function_optimization/Function_2D.ipynb.
I got a TypeError at In[9].

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Cell In[9], [line 7](vscode-notebook-cell:?execution_count=9&line=7)
      [3](vscode-notebook-cell:?execution_count=9&line=3) site_list = torch.arange(len(domain))#len(domain_task)+torch.arange(len(domain_decision))
      [4](vscode-notebook-cell:?execution_count=9&line=4) domain_new = tt_utils.refine_domain(domain=domain, 
      [5](vscode-notebook-cell:?execution_count=9&line=5)                                     site_list=site_list,
      [6](vscode-notebook-cell:?execution_count=9&line=6)                                     scale_factor=scale_factor, device=device)
----> [7](vscode-notebook-cell:?execution_count=9&line=7) tt_model_new = tt_utils.refine_model(tt_model=tt_model, 
      [8](vscode-notebook-cell:?execution_count=9&line=8)                                     site_list=site_list,
      [9](vscode-notebook-cell:?execution_count=9&line=9)                                     scale_factor=scale_factor, device=device)

TypeError: refine_model() got an unexpected keyword argument 'scale_factor'

It seems that the inputs here do not match the code.