Skip to content

Commit 970f5d9

Browse files
committed
backend: new data functions
1 parent 2b0db32 commit 970f5d9

File tree

3 files changed

+31
-3
lines changed

3 files changed

+31
-3
lines changed

pysits/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@
8989
sits_stats,
9090
sits_whittaker,
9191
)
92-
from .sits.utils import get_package_dir, read_sits_rds
92+
from .sits.utils import get_package_dir, load_samples_dataset, r_set_seed, read_sits_rds
9393
from .sits.visualization import sits_plot as plot
9494

9595
__all__ = (
@@ -179,6 +179,8 @@
179179
# Utils
180180
"read_sits_rds",
181181
"get_package_dir",
182+
"load_samples_dataset",
183+
"r_set_seed",
182184
# Package settings
183185
"__version__",
184186
)

pysits/backend/data.py

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,27 @@
2020
from rpy2.robjects import r
2121

2222

23-
def load_global_data(name: str) -> object:
24-
"""Load global data from R environment."""
23+
def load_data_from_package(name: str, package: str, **kwargs) -> object:
24+
"""Load data from package.
25+
26+
This function loads data from a package. It uses `data` behind the scenes.
27+
28+
Args:
29+
name (str): Dataset name.
30+
31+
package (str): Package name.
32+
33+
**kwargs: Additional arguments to pass to the function.
34+
"""
35+
return r.data(name, package=package, **kwargs)
36+
37+
38+
def load_data_from_global(name: str) -> object:
39+
"""Load data from global environment.
40+
41+
This function loads data from the global environment.
42+
43+
Args:
44+
name (str): Dataset name.
45+
"""
2546
return r[name]

pysits/backend/functions.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,11 @@
3939
#
4040
r_fnc_system_file = r_pkg_base.system_file
4141

42+
#
43+
# Base - set.seed
44+
#
45+
r_fnc_set_seed = r_pkg_base.set_seed
46+
4247
#
4348
# Base - Class
4449
#

0 commit comments

Comments
 (0)