Verified Commit b018b4bb authored by Yunus Sevinchan's avatar Yunus Sevinchan
Browse files

Fix deprecated imports in model plots 

Also now uses the is_operation decorator for defining operations.
parent 80788ed3

python/model_plots/SimpleFlocking/__init__.py

+0 −10
Original line number Diff line number Diff line 
"""Plots and operations specific to the SimpleFlocking model"""



from utopya.plotting import register_operation



from .operations import *

from .spatial import agents_in_domain
 


# Register custom operations

register_operation(

    name="set_nan_at_discontinuities", func=set_nan_at_discontinuities

)

register_operation(

    name="insert_nan_at_discontinuities", func=insert_nan_at_discontinuities

)

python/model_plots/SimpleFlocking/operations.py

+4 −0
Original line number Diff line number Diff line
@@ -3,7 +3,10 @@ 
import numpy as np

import xarray as xr



from utopya.eval import is_operation





@is_operation

def set_nan_at_discontinuities(

    ds: xr.Dataset, *, variable: str, threshold: float

) -> xr.Dataset:
@@ -44,6 +47,7 @@ def set_nan_at_discontinuities( 
    )





@is_operation

def insert_nan_at_discontinuities(

    ds: xr.Dataset, *, variable: str, threshold: float

) -> xr.Dataset:

python/model_plots/dummy/__init__.py

+1 −1
Original line number Diff line number Diff line
@@ -2,7 +2,7 @@ 


# Make them all available here to allow easier import

# Register some custom operation with the DAG framework

from utopya.plotting import register_operation

from utopya.eval import register_operation



from .state import *