refactor: rename triton_distributed to dynemo (#22)

Co-authored-by: Graham King <grahamk@nvidia.com>

lib/runtime/tests/common/mock.rs

@@ -21,8 +21,8 @@ use futures::StreamExt;
 use serde::{Deserialize, Serialize};
 use tokio::sync::mpsc;
 
-use triton_distributed_runtime::engine::{AsyncEngine, AsyncEngineContext, Data, ResponseStream};
-use triton_distributed_runtime::pipeline::{
+use dynemo_runtime::engine::{AsyncEngine, AsyncEngineContext, Data, ResponseStream};
+use dynemo_runtime::pipeline::{
     context::{Context, StreamContext},
     Error, ManyOut, PipelineError, PipelineIO, SegmentSource, SingleIn,
 };

lib/runtime/tests/lifecycle.rs

@@ -13,7 +13,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-use triton_distributed_runtime::{worker::Worker, Result, Runtime};
+use dynemo_runtime::{worker::Worker, Result, Runtime};
 
 async fn hello_world(_runtime: Runtime) -> Result<()> {
     Ok(())

lib/runtime/tests/pipeline.rs

@@ -17,8 +17,8 @@ use futures::{stream, StreamExt};
 use serde::{Deserialize, Serialize};
 use std::{sync::Arc, time::Duration};
 
-use triton_distributed_runtime::engine::ResponseStream;
-use triton_distributed_runtime::{
+use dynemo_runtime::engine::ResponseStream;
+use dynemo_runtime::{
     pipeline::{
         async_trait, AsyncEngine, Data, Event, ManyOut, Operator, ServiceBackend, ServiceEngine,
         ServiceFrontend, SingleIn, *,

lib/runtime/tests/soak.rs

@@ -16,12 +16,9 @@
 #[cfg(feature = "integration")]
 mod integration {
 
-    pub const DEFAULT_NAMESPACE: &str = "triton-init";
+    pub const DEFAULT_NAMESPACE: &str = "dynemo";
 
-    use futures::StreamExt;
-    use std::{sync::Arc, time::Duration};
-    use tokio::time::Instant;
-    use triton_distributed_runtime::{
+    use dynemo_runtime::{
         logging,
         pipeline::{
             async_trait, network::Ingress, AsyncEngine, AsyncEngineContextProvider, Error, ManyOut,
@@ -30,6 +27,9 @@ mod integration {
         protocols::annotated::Annotated,
         DistributedRuntime, ErrorContext, Result, Runtime, Worker,
     };
+    use futures::StreamExt;
+    use std::{sync::Arc, time::Duration};
+    use tokio::time::Instant;
 
     #[test]
     fn main() -> Result<()> {
@@ -100,11 +100,11 @@ mod integration {
 
     async fn client(runtime: DistributedRuntime) -> Result<()> {
         // get the run duration from env
-        let run_duration = std::env::var("TRD_SOAK_RUN_DURATION").unwrap_or("1m".to_string());
+        let run_duration = std::env::var("DYN_SOAK_RUN_DURATION").unwrap_or("1m".to_string());
         let run_duration =
             humantime::parse_duration(&run_duration).unwrap_or(Duration::from_secs(60));
 
-        let batch_load = std::env::var("TRD_SOAK_BATCH_LOAD").unwrap_or("10000".to_string());
+        let batch_load = std::env::var("DYN_SOAK_BATCH_LOAD").unwrap_or("10000".to_string());
         let batch_load: usize = batch_load.parse().unwrap_or(10000);
 
         let client = runtime

pyproject.toml

@@ -14,12 +14,12 @@
 # limitations under the License.
 
 [project]
-name = "triton-distributed"
+name = "dynemo"
 version = "0.2.1"
 description = "Distributed Inference Framework"
 readme = "README.md"
 authors = [
-    { name = "NVIDIA Inc.", email = "sw-dl-triton@nvidia.com" },
+    { name = "NVIDIA Inc.", email = "sw-dl-dynemo@nvidia.com" },
 ]
 license = { file = "LICENSE" }
 requires-python = ">=3.10"
@@ -30,11 +30,11 @@ dependencies = [
 ]
 
 [tool.maturin]
-module-name = "triton_distributed._core"
+module-name = "dynemo._core"
 # Point to the Rust crate directory
 manifest-path = "lib/bindings/python/Cargo.toml"
 # Include all Python packages
-python-packages = ["triton_distributed"]
+python-packages = ["dynemo"]
 # Specify the Python source directory
 python-source = "lib/bindings/python/src"
 
@@ -75,7 +75,7 @@ line_length = 88
 balanced_wrapping = true
 indent = "    "
 skip = ["build"]
-known_first_party = ["triton_distributed"]
+known_first_party = ["dynemo"]
 
 [tool.pytest.ini_options]
 minversion = "8.0"