Fixes for TorchRec installation script (#2868)

Summary:
- Fixes for TorchRec installation script

Pull Request resolved: #2868

Reviewed By: sryap

Differential Revision: D59993465

Pulled By: q10

fbshipit-source-id: 195173efb4cdc8c36b242558baf2606c7ee3f970

.github/scripts/setup_env.bash

@@ -24,6 +24,8 @@
 # shellcheck disable=SC1091,SC2128
 . "$( dirname -- "$BASH_SOURCE"; )/utils_triton.bash"
 # shellcheck disable=SC1091,SC2128
+. "$( dirname -- "$BASH_SOURCE"; )/utils_torchrec.bash"
+# shellcheck disable=SC1091,SC2128
 . "$( dirname -- "$BASH_SOURCE"; )/fbgemm_build.bash"
 # shellcheck disable=SC1091,SC2128
 . "$( dirname -- "$BASH_SOURCE"; )/fbgemm_gpu_build.bash"

.github/scripts/utils_torchrec.bash

@@ -45,7 +45,7 @@ install_torchrec_pip () {
   (exec_with_retries 3 conda install ${env_prefix} -c conda-forge -y \
     iopath \
     lightning-utilities \
-    pyre_extensions) || return 1
+    pyre-extensions) || return 1
 
   # Install the package from TorchRec PIP (not PyPI)
   install_from_pytorch_pip "${env_name}" torchrec "${torchrec_channel_version}" "${torchrec_variant_type_version}" || return 1