ModuleNotFoundError: No module named 'nni.algorithms.compression'

[mo-shahab]

Describe the issue:

Environment:

• NNI version: 3.0
• Training service (local|remote|pai|aml|etc):
• Client OS:
• Server OS (for remote mode only):
• Python version: 3.12
• PyTorch/TensorFlow version: Pytorch=2.2.1
• Is conda/virtualenv/venv used?: yes
• Is running in Docker?: no

Configuration:

• Experiment config (remember to remove secrets!):
• Search space:

Log message:

• nnimanager.log:
• dispatcher.log:
• nnictl stdout and stderr:

How to reproduce it?:

[mo-shahab]

import statements in the script

import functools, copy
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from nni.algorithms.compression.pytorch.pruning import L1FilterPruner, L1FilterPrunerMasker

from nni.algorithms.compression.pytorch.pruning import L1FilterPruner, L1FilterPrunerMasker

loc giving the error