File size: 541 Bytes
3883c60
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from setup_tools.magicinstaller.requirement import Requirement, SimpleRequirement


class AudioLM(Requirement):
    def is_right_version(self):
        return self.get_package_version('audiolm-pytorch') == '1.1.4'

    def is_installed(self):
        return self.install_check('audiolm-pytorch')

    def install(self) -> tuple[int, str, str]:
        return self.install_pip('audiolm-pytorch==1.1.4', 'audiolm')


class JobLib(SimpleRequirement):
    package_name = 'joblib'


class FairSeq(SimpleRequirement):
    package_name = 'fairseq'